QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#101352 | #5031. 核 | zhoukangyang | 60 | 261ms | 28428kb | C++17 | 4.3kb | 2023-04-29 11:01:12 | 2023-04-29 11:01:15 |
Judging History
answer
#include<bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native")
#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 = 1e7 + 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)
fac[i] = (ll) fac[i - 1] * fac[i] % mod;
ifac[x] = qpow(fac[x]);
R(i, x - 1, 0)
ifac[i] = (ll) ifac[i + 1] * (pw[i + 1] - 1) % 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;
int f[N], g[N], ans[N], T[N], suf[N];
int Pw1[N], Pw2[N];
int main() {
ios :: sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> n >> q >> mod;
// n = 1e7, q = 998244353, mod = 1e9 + 7;
q_init(n + 1);
// cout << clock() << endl;
int mn = min(n, 80);
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;
}
}
L(i, 0, n) {
ans[i] = (ll) ans[i] * q_C(n, i) % mod;
}
if(mn < n) S.gauss(ans, mn);
L(i, mn + 1, n) {
int pw = 1;
unsigned long long t = 0;
L(y, 0, 3) t += (ll) pw * S.f[0][y], pw = (ll) pw * ::pw[i] % mod;
T[i] = t % mod;
}
suf[n + 1] = 1;
R(i, n, mn + 1)
suf[i] = (ll) suf[i + 1] * T[i] % mod;
int all = qpow(suf[mn + 1]);
L(i, mn + 1, n) {
unsigned long long s = 0;
L(x, 1, 2) {
int pw = ans[i - x];
s += (ll) pw * S.f[x][0];
pw = (ll) pw * ::pw[i] % mod;
s += (ll) pw * S.f[x][1];
pw = (ll) pw * ::pw[i] % mod;
s += (ll) pw * S.f[x][2];
pw = (ll) pw * ::pw[i] % mod;
s += (ll) pw * S.f[x][3];
}
ans[i] = (ll) (mod - s % mod) * all % mod * suf[i + 1] % mod;
all = (ll) all * T[i] % mod;
}
int pw = 1;
L(i, 1, n) {
pw = (ll) pw * q % (mod - 1);
}
const int K = sqrt(mod) + 1;
Pw1[0] = Pw2[0] = 1;
L(i, 1, K) {
Pw1[i] = (ll) Pw1[i - 1] * 3 % mod;
}
L(i, 1, K) {
Pw2[i] = (ll) Pw2[i - 1] * Pw1[K] % mod;
}
int s = 1;
ull ns = 0;
R(i, n, 0) {
int w = (ll) Pw1[s % K] * Pw2[s / K] % mod;
ns += (ull) ans[i] * w;
if(!(i & 16)) ns %= mod;
s = (ll) s * pw % (mod - 1);
}
ns %= mod;
cout << ns << '\n';
return 0;
}
詳細信息
Subtask #1:
score: 10
Accepted
Test #1:
score: 10
Accepted
time: 3ms
memory: 5784kb
input:
1 2 540053233
output:
9
result:
ok 1 number(s): "9"
Test #2:
score: 0
Accepted
time: 3ms
memory: 5504kb
input:
2 2 156542707
output:
43046970
result:
ok 1 number(s): "43046970"
Test #3:
score: 0
Accepted
time: 2ms
memory: 3652kb
input:
1 2 186225229
output:
9
result:
ok 1 number(s): "9"
Test #4:
score: 0
Accepted
time: 1ms
memory: 3480kb
input:
3 3 109884329
output:
100602209
result:
ok 1 number(s): "100602209"
Test #5:
score: 0
Accepted
time: 2ms
memory: 3548kb
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: 3524kb
input:
20 21992843 328859143
output:
110137213
result:
ok 1 number(s): "110137213"
Test #7:
score: 0
Accepted
time: 0ms
memory: 3744kb
input:
22 332524739 654888401
output:
410922781
result:
ok 1 number(s): "410922781"
Test #8:
score: 0
Accepted
time: 0ms
memory: 3580kb
input:
26 302215049 566649113
output:
221720840
result:
ok 1 number(s): "221720840"
Test #9:
score: 0
Accepted
time: 1ms
memory: 3656kb
input:
15 111009527 722130737
output:
648834664
result:
ok 1 number(s): "648834664"
Test #10:
score: 0
Accepted
time: 3ms
memory: 3608kb
input:
82 110032063 394529383
output:
111730592
result:
ok 1 number(s): "111730592"
Test #11:
score: 0
Accepted
time: 2ms
memory: 3564kb
input:
9 11172911 259650437
output:
68381774
result:
ok 1 number(s): "68381774"
Test #12:
score: 0
Accepted
time: 3ms
memory: 3656kb
input:
86 12016027 354886243
output:
263687778
result:
ok 1 number(s): "263687778"
Test #13:
score: 0
Accepted
time: 3ms
memory: 3728kb
input:
91 273689959 454097881
output:
114436127
result:
ok 1 number(s): "114436127"
Test #14:
score: 0
Accepted
time: 0ms
memory: 3588kb
input:
73 148878967 694206977
output:
176215101
result:
ok 1 number(s): "176215101"
Test #15:
score: 0
Accepted
time: 2ms
memory: 3504kb
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: 3ms
memory: 3676kb
input:
2778 122825869 147297463
output:
43419574
result:
ok 1 number(s): "43419574"
Test #17:
score: 0
Accepted
time: 1ms
memory: 3704kb
input:
289 7729669 589652893
output:
552952137
result:
ok 1 number(s): "552952137"
Test #18:
score: 0
Accepted
time: 3ms
memory: 3524kb
input:
2281 35651417 203950963
output:
21659018
result:
ok 1 number(s): "21659018"
Test #19:
score: 0
Accepted
time: 3ms
memory: 3660kb
input:
1684 258745639 373223677
output:
355596229
result:
ok 1 number(s): "355596229"
Test #20:
score: 0
Accepted
time: 1ms
memory: 3736kb
input:
2107 86850989 455823859
output:
245960059
result:
ok 1 number(s): "245960059"
Test #21:
score: 0
Accepted
time: 4ms
memory: 3564kb
input:
1323 43290799 791120419
output:
509649562
result:
ok 1 number(s): "509649562"
Test #22:
score: 0
Accepted
time: 0ms
memory: 3536kb
input:
2401 34064903 185314627
output:
70571452
result:
ok 1 number(s): "70571452"
Test #23:
score: 0
Accepted
time: 3ms
memory: 3616kb
input:
1073 82288187 564447959
output:
168200843
result:
ok 1 number(s): "168200843"
Test #24:
score: 0
Accepted
time: 3ms
memory: 3540kb
input:
1926 29995039 129122281
output:
60921463
result:
ok 1 number(s): "60921463"
Test #25:
score: 0
Accepted
time: 1ms
memory: 3800kb
input:
3000 66915659 765705179
output:
222619979
result:
ok 1 number(s): "222619979"
Subtask #4:
score: 20
Accepted
Test #26:
score: 20
Accepted
time: 261ms
memory: 27120kb
input:
998818 198334853 998244353
output:
153251445
result:
ok 1 number(s): "153251445"
Test #27:
score: 0
Accepted
time: 246ms
memory: 25192kb
input:
914379 128814383 998244353
output:
477606145
result:
ok 1 number(s): "477606145"
Test #28:
score: 0
Accepted
time: 248ms
memory: 25832kb
input:
944474 478445339 998244353
output:
174204073
result:
ok 1 number(s): "174204073"
Test #29:
score: 0
Accepted
time: 256ms
memory: 28428kb
input:
948637 711592127 998244353
output:
178256114
result:
ok 1 number(s): "178256114"
Test #30:
score: 0
Accepted
time: 239ms
memory: 26100kb
input:
927564 14465663 998244353
output:
315244613
result:
ok 1 number(s): "315244613"
Test #31:
score: 0
Accepted
time: 243ms
memory: 26324kb
input:
934615 392799073 998244353
output:
892700270
result:
ok 1 number(s): "892700270"
Test #32:
score: 0
Accepted
time: 233ms
memory: 27868kb
input:
917196 124972031 998244353
output:
782017412
result:
ok 1 number(s): "782017412"
Test #33:
score: 0
Accepted
time: 249ms
memory: 26956kb
input:
957149 392606173 998244353
output:
159348443
result:
ok 1 number(s): "159348443"
Test #34:
score: 0
Accepted
time: 261ms
memory: 27688kb
input:
997042 184649453 998244353
output:
464643024
result:
ok 1 number(s): "464643024"
Test #35:
score: 0
Accepted
time: 250ms
memory: 28404kb
input:
953353 14071961 998244353
output:
391688875
result:
ok 1 number(s): "391688875"
Subtask #5:
score: 0
Time Limit Exceeded
Test #36:
score: 0
Time Limit Exceeded
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%