QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#582718 | #9380. Match | arahato | AC ✓ | 157ms | 4220kb | C++14 | 2.6kb | 2024-09-22 17:15:24 | 2024-09-22 17:15:31 |
Judging History
answer
#include <bits/stdc++.h>
#include <iostream>
using namespace std;
#define int long long
const int N = 205, mod = 998244353;
int n, k, C[N][N];
signed main(){
ios::sync_with_stdio(0);
cin >> n >> k;
vector<int> a(n), b(n), fac(n + 1);
fac[0] = 1;
for (int i = 1; i <= n; i++){
fac[i] = fac[i - 1] * i % mod;
}
for (int i = 0; i < n; i++){
cin >> a[i];
}
for (int i = 0; i < n; i++){
cin >> b[i];
}
for (int i = 0; i <= n; i++){
C[i][0] = 1;
for (int j = 1; j <= i; j++){
C[i][j] = (C[i - 1][j] + C[i - 1][j - 1]) % mod;
}
}
function<vector<int>(vector<int>&, vector<int>&, int)> solve
= [&](auto &a, auto &b, int p) -> vector<int> {
if (a.empty() || b.empty()){
return {1};
}
if (p < 0) {
vector<int> res(min(a.size(), b.size()) + 1);
for (int i = 0; i < res.size(); i++){
res[i] = C[a.size()][i] * C[b.size()][i] % mod * fac[i] % mod;
}
return res;
}
vector<int> a0, a1, b0, b1;
for (int x : a){
if (x >> p & 1){
a1.push_back(x);
} else {
a0.push_back(x);
}
}
for (int x : b){
if (x >> p & 1){
b1.push_back(x);
} else {
b0.push_back(x);
}
}
auto combine = [&](auto &x, auto &y) -> vector<int> {
vector<int> res(x.size() + y.size() + 1);
for (int i = 0; i < x.size(); i++){
for (int j = 0; j < y.size(); j++){
res[i + j] = (res[i + j] + x[i] * y[j]) % mod;
}
}
while (res.back() == 0) res.pop_back();
return res;
};
if (k >> p & 1){
auto x = solve(a1, b0, p - 1), y = solve(a0, b1, p - 1);
return combine(x, y);
} else {
auto A = solve(a0, b0, p - 1), B = solve(a1, b1, p - 1);
vector<int> res(min(a.size(), b.size()) + 1);
for (int i = 0; i < A.size(); i++){
for (int j = 0; j < B.size(); j++){
int restx = min(a0.size() - i, b1.size() - j), resty = min(b0.size() - i, a1.size() - j);
for (int x = 0; x <= restx; x++) for (int y = 0; y <= resty; y++){
res[i + j + x + y] = (res[i + j + x + y] + A[i] * B[j] % mod
* C[a0.size() - i][x] % mod * C[b1.size() - j][x] % mod * fac[x] % mod
* C[b0.size() - i][y] % mod * C[a1.size() - j][y] % mod * fac[y]) % mod;
}
}
}
while (res.back() == 0) res.pop_back();
return res;
}
};
auto ans = solve(a, b, 60);
while (ans.size() <= n) ans.push_back(0);
for (int i = 1; i < ans.size(); i++){
cout << ans[i] << endl;
}
}
这程序好像有点Bug,我给组数据试试?
詳細信息
Test #1:
score: 100
Accepted
time: 1ms
memory: 3556kb
input:
9 5 1 7 2 8 4 9 2 5 10 1 3 2 4 5 8 8 8 9
output:
51 1034 10768 62195 200965 348924 294444 97344 7200
result:
ok 9 lines
Test #2:
score: 0
Accepted
time: 157ms
memory: 4012kb
input:
200 569102225177443347 1103663383682482176 1103381908705771520 1099441259031822336 1098878309078401024 1089871109823660032 1080863910568919040 1074108511127863296 1071856711314178048 1069041961547071488 1068479011593650176 1067353111686807552 1062849512059437056 1049338713177325568 10448351135499550...
output:
20506 208107723 47878304 53020813 972282728 933586852 658157196 670189811 957980024 366179738 217980591 967482558 833450149 987731802 260904367 5263881 600332344 906061351 658256294 93700706 421323952 178075016 219871690 986880524 848776106 191185484 641917326 576497440 908609746 349728876 606714342...
result:
ok 200 lines
Test #3:
score: 0
Accepted
time: 10ms
memory: 3960kb
input:
200 1098723432450995412 972777519512027136 936748722493063168 864691128455135232 860187528827764736 857935729014079488 856246879153815552 855683929200394240 851180329573023744 848928529759338496 847802629852495872 847310048643252224 847239679899074560 846676729945653248 828662331436171264 8280993814...
output:
8320 34035512 428014253 916072411 696504298 748377440 32424677 188402417 233587075 130639672 759476380 546285625 187068736 159002787 131866334 381530906 133126344 373727612 923311054 725293681 718162548 39511135 322320638 44709653 156884882 285926751 787179409 282967016 153092344 615721347 950855850...
result:
ok 200 lines
Test #4:
score: 0
Accepted
time: 1ms
memory: 4156kb
input:
200 1 1 1 1 1 1 1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...
output:
3072 4412928 940025853 665370743 752672705 581549490 41990996 541401698 170451802 26584141 220983766 844620126 64506869 137621326 418866920 351049174 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...
result:
ok 200 lines
Test #5:
score: 0
Accepted
time: 2ms
memory: 4220kb
input:
200 0 1 1 1 1 1 1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...
output:
40000 792020000 319908096 286630939 515085548 95626140 472749362 800008042 594973486 139901984 847967250 125664877 254232250 419486325 204481923 979660340 41873095 540955890 585129047 205968776 196086667 674391130 737319172 669580034 973843015 207125612 331414909 119844754 24879477 990932807 8476927...
result:
ok 200 lines
Extra Test:
score: 0
Extra Test Passed