QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#449315 | #7838. 往日之影 | Made_in_Code | 0 | 49ms | 42716kb | C++14 | 3.7kb | 2024-06-20 22:14:24 | 2024-06-20 22:14:25 |
Judging History
answer
#include <iostream>
#define LL long long
using namespace std;
const int kMaxN = 1e6 + 1, kMaxL = 20;
struct H {
int pw;
LL fact, _fact;
} h[kMaxN];
int t, p, n, a[4], d[4][4];
void Add(LL &x, LL y) { x = (x + y) % p; }
LL Pow(LL x, LL y) {
LL ans = 1;
for (LL i = 1; i <= y; i <<= 1) {
if (i & y) {
ans = ans * x % p;
}
x = x * x % p;
}
return ans;
}
class P {
public:
LL r, i;
P() : r(0), i(0) {}
P(LL _r, LL _i) : r(_r), i(_i) {}
P operator+(P x) {
return {(r + x.r) % p, (i + x.i) % p};
}
P operator<<(LL x) {
if ((x & 3) == 0) {
return P(r, i);
} else if ((x & 3) == 1) {
return P(-i, r);
} else if ((x & 3) == 2) {
return P(-r, -i);
} else {
return P(i, -r);
}
}
P operator*(LL x) {
return {r * x % p, i * x % p};
}
P operator*(P x) {
return {(r * x.r - i * x.i) % p, (r * x.i + i * x.r) % p};
}
P operator^(LL y) {
P x = *this, ans(1, 0);
for (LL i = 1; i <= y; i <<= 1) {
if (i & y) {
ans = ans * x;
}
x = x * x;
}
return ans;
}
} ans;
void CalcFact() {
int _r;
LL pw[kMaxL], s[kMaxN], t[kMaxN];
pw[0] = 1;
for (int &i = _r = 1; i < kMaxL; i++) {
pw[i] = pw[i - 1] * p;
if (pw[i] >= kMaxN) {
break;
}
}
h[0] = {0, 1, 1};
for (int i = 1; i < kMaxN; i++) {
int l = 0, r = _r;
while (l <= r) {
int mid = l + r >> 1;
if (mid <= _r && !(i % pw[mid])) {
l = mid + 1;
} else {
r = mid - 1;
}
}
h[i] = h[i - 1], h[i].pw += r;
h[i].fact = h[i].fact * (i / pw[r]) % p;
}
s[0] = 1;
for (int i = 1; i < kMaxN; i++) {
s[i] = s[i - 1] * h[i].fact % p;
}
t[kMaxN - 1] = Pow(s[kMaxN - 1], p - 2);
for (int i = kMaxN - 1; i > 0; i--) {
t[i - 1] = t[i] * h[i].fact % p;
}
for (int i = 1; i < kMaxN; i++) {
h[i]._fact = s[i - 1] * t[i] % p;
}
}
void CalcAns(int o) {
LL b[4] = {};
P w(o, 0);
for (int i = 0; i < 4; i++) {
for (int j = 0; j < 4; j++) {
b[j] += d[i][j];
}
}
if (!b[0] && !b[2]) {
return;
}
for (int i = 0; i < 4; i++) {
w.r = w.r * h[a[i]].fact % p;
for (int j = 0; j < 4; j++) {
w.r = w.r * h[d[i][j]]._fact % p;
}
}
int q = (d[1][3] + d[3][1]) + 3 * (d[1][1] + d[3][3]) & 3;
q = q + 2 * (d[1][2] + d[2][1] + d[2][3] + d[3][2]) & 3;
w = w << q;
for (int i = 0; i < 4; i++) {
for (int j = i + 1; j < 4; j++) {
P p(1, 0);
if (i + j & 1) {
p.i += i + j & 2 ? -1 : 1;
} else {
p.r += i + j & 2 ? -1 : 1;
}
w = w * (p ^ b[i] * b[j]);
}
}
for (int i = 0; i < 4; i++) {
P p(1, 0);
p.r += i & 1 ? -1 : 1;
w = w * (p ^ b[i] * (b[i] - 1) / 2);
}
ans = ans + w;
}
void S(int x, int b0, int b1, int b2, int b3, int o) {
if (x == 4) {
CalcAns(o);
return;
}
for (int &i = d[x][0] = 0; i <= b0 && i <= a[x]; i++) {
for (int &j = d[x][1] = 0; j <= b1 && i + j <= a[x]; j++) {
for (int &k = d[x][2] = 0; k <= b2 && i + j + k <= a[x]; k++) {
int &l = d[x][3] = a[x] - i - j - k;
if (l <= b3 && h[i].pw + h[j].pw + h[k].pw + h[l].pw == h[a[x]].pw) {
S(x + 1, b0 - i, b1 - j, b2 - k, b3 - l, o);
}
}
}
}
}
int main() {
cin.tie(0), cout.tie(0);
ios::sync_with_stdio(0);
cin >> t >> p;
CalcFact();
while (t--) {
cin >> n >> a[0] >> a[1] >> a[2] >> a[3];
ans = P();
S(0, 0, 1, n, 1, 1);
S(0, n, 1, 0, 1, 1);
S(0, 0, 1, 0, 1, -1);
int q = p - 1 - 1LL * n * (n + 3) / 2 % (p - 1);
cout << (ans.r + p) * Pow(2, q) % p << '\n';
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 0
Wrong Answer
Test #1:
score: 10
Accepted
time: 30ms
memory: 42580kb
input:
1 998244353 3 1 1 0 1
output:
0
result:
ok single line: '0'
Test #2:
score: 0
Accepted
time: 33ms
memory: 42636kb
input:
1 998244353 7 0 2 1 4
output:
998069185
result:
ok single line: '998069185'
Test #3:
score: 0
Accepted
time: 30ms
memory: 42716kb
input:
1 998244353 4 0 1 0 3
output:
0
result:
ok single line: '0'
Test #4:
score: -10
Wrong Answer
time: 29ms
memory: 42704kb
input:
1 998244353 2 1 0 1 0
output:
873463809
result:
wrong answer 1st lines differ - expected: '0', found: '873463809'
Subtask #2:
score: 0
Skipped
Dependency #1:
0%
Subtask #3:
score: 0
Skipped
Dependency #2:
0%
Subtask #4:
score: 0
Wrong Answer
Test #23:
score: 0
Wrong Answer
time: 49ms
memory: 42648kb
input:
999 999999001 2 2 0 0 0 3 3 0 0 0 4 4 0 0 0 5 5 0 0 0 6 6 0 0 0 7 7 0 0 0 8 8 0 0 0 9 9 0 0 0 10 10 0 0 0 11 11 0 0 0 12 12 0 0 0 13 13 0 0 0 14 14 0 0 0 15 15 0 0 0 16 16 0 0 0 17 17 0 0 0 18 18 0 0 0 19 19 0 0 0 20 20 0 0 0 21 21 0 0 0 22 22 0 0 0 23 23 0 0 0 24 24 0 0 0 25 25 0 0 0 26 26 0 0 0 27...
output:
624999376 874999126 359374641 919920956 691222454 586081873 33512082 496961574 790501684 206445579 708073277 492142887 486007979 21786019 802052117 198521403 854660059 658779344 904643630 538486221 357736277 949763680 94144464 342842045 695164947 276856011 552666277 813428208 572457238 910726512 177...
result:
wrong answer 1st lines differ - expected: '499999501', found: '624999376'
Subtask #5:
score: 0
Skipped
Dependency #1:
0%