QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#101341 | #5031. 核 | zhoukangyang | 60 | 1856ms | 22996kb | C++17 | 3.5kb | 2023-04-29 10:42:15 | 2023-04-29 10:42:17 |
Judging History
answer
#include<bits/stdc++.h>
#define L(i, j, k) for(int i = (j); i <= (k); ++i)
#define R(i, j, k) for(int i = (j); i >= (k); --i)
#define ll long long
#define vi vector <int>
#define sz(a) ((int) (a).size())
#define me(f, x) memset(f, x, sizeof(f))
#define uint unsigned int
#define ull unsigned long long
#define bs bitset < N >
using namespace std;
const int N = 1e6 + 7;
int n, q, mod;
int pw[N], fac[N], ifac[N];
int qpow(int x, int y = mod - 2) {
int res = 1;
for(; y; x = (ll) x * x % mod, y >>= 1) if(y & 1) res = (ll) res * x % mod;
return res;
}
void q_init(int x) {
pw[0] = 1;
L(i, 1, x) pw[i] = (ll) pw[i - 1] * q % mod;
L(i, 1, x) fac[i] = pw[i] - 1;
fac[0] = ifac[0] = 1;
L(i, 1, x)
ifac[i] = (ll) qpow (fac[i]) * ifac[i - 1] % mod,
fac[i] = (ll) fac[i - 1] * fac[i] % mod;
}
int q_C(int x, int y) {
return x < y || y < 0 ? 0 : (ll) fac[x] * ifac[y] % mod * ifac[x - y] % mod;
}
inline int sgn(int x) {
return (x & 1) ? mod - 1 : 1;
}
template < int L, int K > struct DBM {
static const int R = (L + 1) * (K + 1);
int lpos, a[R * 10 + 7];
int G[R + 7][R + 7], cur[R + 7], ns[R + 7];
int f[L + 2][K + 2], id[L + 2][K + 2], idt;
inline void insert() {
R(i, idt, 1) if(cur[i]) {
if(!G[i][i]) {
L(j, 1, i)
G[i][j] = cur[j];
}
int iv = (ll) (mod - cur[i]) * qpow(G[i][i]) % mod;
L(j, 1, i)
(cur[j] += (ll) iv * G[i][j] % mod) %= mod;
}
}
void gauss(int *a, int n, int lpos = 0) {
n = min(n, R * 10), idt = 0;
L(i, 0, L)
L(j, 0, K)
id[i][j] = ++idt;
L(i, L + 1, n) {
L(j, 0, idt)
cur[j] = 0;
L(x, 0, L) if(x < i) {
int pw = a[i - x];
L(y, 0, K)
cur[id[x][y]] = pw,
pw = (ll) pw * ::pw[lpos + i] % mod;
}
insert();
}
int pos = 1;
while(G[pos][pos]) ++pos;
ns[pos] = 1;
if(pos > idt) {
cout << "TAT" << endl;
exit(0);
}
L(i, pos + 1, idt) {
int s = 0;
L(j, 1, i - 1) {
(s += (ll) G[i][j] * ns[j] % mod) %= mod;
}
ns[i] = (ll) (mod - s) * qpow(G[i][i]) % mod;
}
L(i, 0, L)
L(j, 0, K)
f[i][j] = ns[id[i][j]];
L(i, L + 1, n) {
L(j, 0, idt)
cur[j] = 0;
int s = 0;
L(x, 0, L) if(x < i) {
int pw = a[i - x];
L(y, 0, K)
(s += (ll) pw * f[x][y] % mod) %= mod,
pw = (ll) pw * ::pw[lpos + i] % mod;
}
}
}
};
DBM < 2, 3 > S;
const int T = 80;
int f[N], g[N], ans[N];
int main() {
ios :: sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> n >> q >> mod;
q_init(n + 1);
int mn = min(n, T);
f[0] = 1;
L(i, 1, mn) {
f[i] = (ll) f[i - 1] * (pw[n] + mod - pw[n - i]) % mod;
}
L(i, 0, mn) {
g[i] = (ll) sgn(i) * qpow(q, i * (i - 1) / 2) % mod;
}
L(i, 0, mn) {
L(j, 0, mn - i) {
(ans[i + j] += (ll) q_C(i + j, i) * f[i] % mod * g[j] % mod) %= mod;
}
}
S.gauss(ans, mn);
L(i, T + 1, n) {
int s = 0;
L(x, 1, 2) {
int pw = ans[i - x];
L(y, 0, 3)
(s += (ll) pw * S.f[x][y] % mod) %= mod, pw = (ll) pw * ::pw[i] % mod;
}
int fs = 0, pw = 1;
L(y, 0, 3)
(fs += (ll) pw * S.f[0][y] % mod) %= mod, pw = (ll) pw * ::pw[i] % mod;
ans[i] = (ll) (mod - s) * qpow(fs) % mod;
}
L(i, 0, n) {
f[i] = ans[i];
}
L(i, 0, n) {
f[i] = (ll) f[i] * q_C(n, i) % mod;
}
int pw = 1;
L(i, 1, n) {
pw = (ll) pw * q % (mod - 1);
}
int s = 1, ns = 0;
R(i, n, 0) {
(ns += (ll) f[i] * qpow(3, s) % mod) %= mod;
s = (ll) s * pw % (mod - 1);
}
cout << ns << '\n';
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 10
Accepted
Test #1:
score: 10
Accepted
time: 2ms
memory: 3424kb
input:
1 2 540053233
output:
9
result:
ok 1 number(s): "9"
Test #2:
score: 0
Accepted
time: 0ms
memory: 3496kb
input:
2 2 156542707
output:
43046970
result:
ok 1 number(s): "43046970"
Test #3:
score: 0
Accepted
time: 2ms
memory: 3376kb
input:
1 2 186225229
output:
9
result:
ok 1 number(s): "9"
Test #4:
score: 0
Accepted
time: 2ms
memory: 3436kb
input:
3 3 109884329
output:
100602209
result:
ok 1 number(s): "100602209"
Test #5:
score: 0
Accepted
time: 2ms
memory: 3488kb
input:
1 2 144802297
output:
9
result:
ok 1 number(s): "9"
Subtask #2:
score: 20
Accepted
Dependency #1:
100%
Accepted
Test #6:
score: 20
Accepted
time: 2ms
memory: 3536kb
input:
20 21992843 328859143
output:
110137213
result:
ok 1 number(s): "110137213"
Test #7:
score: 0
Accepted
time: 2ms
memory: 3388kb
input:
22 332524739 654888401
output:
410922781
result:
ok 1 number(s): "410922781"
Test #8:
score: 0
Accepted
time: 2ms
memory: 3348kb
input:
26 302215049 566649113
output:
221720840
result:
ok 1 number(s): "221720840"
Test #9:
score: 0
Accepted
time: 2ms
memory: 3376kb
input:
15 111009527 722130737
output:
648834664
result:
ok 1 number(s): "648834664"
Test #10:
score: 0
Accepted
time: 2ms
memory: 3372kb
input:
82 110032063 394529383
output:
111730592
result:
ok 1 number(s): "111730592"
Test #11:
score: 0
Accepted
time: 0ms
memory: 3432kb
input:
9 11172911 259650437
output:
68381774
result:
ok 1 number(s): "68381774"
Test #12:
score: 0
Accepted
time: 2ms
memory: 3300kb
input:
86 12016027 354886243
output:
263687778
result:
ok 1 number(s): "263687778"
Test #13:
score: 0
Accepted
time: 1ms
memory: 3428kb
input:
91 273689959 454097881
output:
114436127
result:
ok 1 number(s): "114436127"
Test #14:
score: 0
Accepted
time: 0ms
memory: 3536kb
input:
73 148878967 694206977
output:
176215101
result:
ok 1 number(s): "176215101"
Test #15:
score: 0
Accepted
time: 0ms
memory: 3348kb
input:
45 205982233 227598247
output:
156769598
result:
ok 1 number(s): "156769598"
Subtask #3:
score: 10
Accepted
Dependency #2:
100%
Accepted
Test #16:
score: 10
Accepted
time: 9ms
memory: 3540kb
input:
2778 122825869 147297463
output:
43419574
result:
ok 1 number(s): "43419574"
Test #17:
score: 0
Accepted
time: 3ms
memory: 3388kb
input:
289 7729669 589652893
output:
552952137
result:
ok 1 number(s): "552952137"
Test #18:
score: 0
Accepted
time: 3ms
memory: 3488kb
input:
2281 35651417 203950963
output:
21659018
result:
ok 1 number(s): "21659018"
Test #19:
score: 0
Accepted
time: 5ms
memory: 3460kb
input:
1684 258745639 373223677
output:
355596229
result:
ok 1 number(s): "355596229"
Test #20:
score: 0
Accepted
time: 2ms
memory: 3480kb
input:
2107 86850989 455823859
output:
245960059
result:
ok 1 number(s): "245960059"
Test #21:
score: 0
Accepted
time: 2ms
memory: 3320kb
input:
1323 43290799 791120419
output:
509649562
result:
ok 1 number(s): "509649562"
Test #22:
score: 0
Accepted
time: 3ms
memory: 3424kb
input:
2401 34064903 185314627
output:
70571452
result:
ok 1 number(s): "70571452"
Test #23:
score: 0
Accepted
time: 5ms
memory: 3556kb
input:
1073 82288187 564447959
output:
168200843
result:
ok 1 number(s): "168200843"
Test #24:
score: 0
Accepted
time: 5ms
memory: 3380kb
input:
1926 29995039 129122281
output:
60921463
result:
ok 1 number(s): "60921463"
Test #25:
score: 0
Accepted
time: 5ms
memory: 3596kb
input:
3000 66915659 765705179
output:
222619979
result:
ok 1 number(s): "222619979"
Subtask #4:
score: 20
Accepted
Test #26:
score: 20
Accepted
time: 1856ms
memory: 22996kb
input:
998818 198334853 998244353
output:
153251445
result:
ok 1 number(s): "153251445"
Test #27:
score: 0
Accepted
time: 1629ms
memory: 21236kb
input:
914379 128814383 998244353
output:
477606145
result:
ok 1 number(s): "477606145"
Test #28:
score: 0
Accepted
time: 1745ms
memory: 21828kb
input:
944474 478445339 998244353
output:
174204073
result:
ok 1 number(s): "174204073"
Test #29:
score: 0
Accepted
time: 1713ms
memory: 22056kb
input:
948637 711592127 998244353
output:
178256114
result:
ok 1 number(s): "178256114"
Test #30:
score: 0
Accepted
time: 1622ms
memory: 21460kb
input:
927564 14465663 998244353
output:
315244613
result:
ok 1 number(s): "315244613"
Test #31:
score: 0
Accepted
time: 1637ms
memory: 21624kb
input:
934615 392799073 998244353
output:
892700270
result:
ok 1 number(s): "892700270"
Test #32:
score: 0
Accepted
time: 1609ms
memory: 21292kb
input:
917196 124972031 998244353
output:
782017412
result:
ok 1 number(s): "782017412"
Test #33:
score: 0
Accepted
time: 1681ms
memory: 22240kb
input:
957149 392606173 998244353
output:
159348443
result:
ok 1 number(s): "159348443"
Test #34:
score: 0
Accepted
time: 1768ms
memory: 22908kb
input:
997042 184649453 998244353
output:
464643024
result:
ok 1 number(s): "464643024"
Test #35:
score: 0
Accepted
time: 1682ms
memory: 22120kb
input:
953353 14071961 998244353
output:
391688875
result:
ok 1 number(s): "391688875"
Subtask #5:
score: 0
Runtime Error
Test #36:
score: 0
Runtime Error
input:
9909956 720431399 720431401
output:
result:
Subtask #6:
score: 0
Skipped
Dependency #1:
100%
Accepted
Dependency #2:
100%
Accepted
Dependency #3:
100%
Accepted
Dependency #4:
100%
Accepted
Dependency #5:
0%