QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#348210 | #8339. Rooted Tree | ucup-team206# | AC ✓ | 198ms | 159912kb | C++17 | 1015b | 2024-03-09 17:26:48 | 2024-03-09 17:26:50 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
const int N = 1e7 + 10;
const int p = 1e9 + 9;
int m, k;
int h[N], hinv[N], hfac[N], g[N], f[N];
inline int Qpow(int x, int y = p - 2) {
int ans = 1;
for (; y; y >>= 1, x = 1ll * x * x % p)
if (y & 1) ans = 1ll * ans * x % p;
return ans;
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0);
cin >> m >> k;
h[0] = 1; hfac[0] = 1;
for (int i = 1; i <= k; ++i) {
h[i] = (h[i - 1] + m - 1) % p;
hfac[i] = 1ll * hfac[i - 1] * h[i] % p;
}
hinv[0] = 1;
for (int i = k, val = Qpow(hfac[k]); i; --i) {
hinv[i] = 1ll * hfac[i - 1] * val % p;
val = 1ll * val * h[i] % p;
}
int ans = 0;
for (int i = 1; i <= k; ++i) {
int val = 1ll * g[i - 1] * hinv[i - 1] % p;
ans = (ans + 1ll * (val + 1) * m) % p;
g[i] = (g[i - 1] - val + 1ll * (val + 1) * m) % p;
}
if (ans < 0) ans += p;
cout << ans << endl;
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 7744kb
input:
6 2
output:
18
result:
ok 1 number(s): "18"
Test #2:
score: 0
Accepted
time: 0ms
memory: 3572kb
input:
2 6
output:
600000038
result:
ok 1 number(s): "600000038"
Test #3:
score: 0
Accepted
time: 16ms
memory: 13208kb
input:
83 613210
output:
424200026
result:
ok 1 number(s): "424200026"
Test #4:
score: 0
Accepted
time: 121ms
memory: 108564kb
input:
48 6713156
output:
198541581
result:
ok 1 number(s): "198541581"
Test #5:
score: 0
Accepted
time: 1ms
memory: 3668kb
input:
1 111
output:
6216
result:
ok 1 number(s): "6216"
Test #6:
score: 0
Accepted
time: 148ms
memory: 117760kb
input:
28 7304152
output:
457266679
result:
ok 1 number(s): "457266679"
Test #7:
score: 0
Accepted
time: 80ms
memory: 67756kb
input:
38 4101162
output:
232117382
result:
ok 1 number(s): "232117382"
Test #8:
score: 0
Accepted
time: 197ms
memory: 158712kb
input:
51 9921154
output:
340670552
result:
ok 1 number(s): "340670552"
Test #9:
score: 0
Accepted
time: 35ms
memory: 31732kb
input:
79 1801157
output:
620550406
result:
ok 1 number(s): "620550406"
Test #10:
score: 0
Accepted
time: 107ms
memory: 90096kb
input:
22 5417157
output:
457449071
result:
ok 1 number(s): "457449071"
Test #11:
score: 0
Accepted
time: 54ms
memory: 54316kb
input:
25 3210162
output:
36368303
result:
ok 1 number(s): "36368303"
Test #12:
score: 0
Accepted
time: 58ms
memory: 50064kb
input:
67 2919160
output:
935195555
result:
ok 1 number(s): "935195555"
Test #13:
score: 0
Accepted
time: 183ms
memory: 139464kb
input:
77 8613163
output:
482832472
result:
ok 1 number(s): "482832472"
Test #14:
score: 0
Accepted
time: 198ms
memory: 159792kb
input:
90 10000000
output:
275581651
result:
ok 1 number(s): "275581651"
Test #15:
score: 0
Accepted
time: 190ms
memory: 159912kb
input:
99 9999999
output:
126087169
result:
ok 1 number(s): "126087169"
Test #16:
score: 0
Accepted
time: 186ms
memory: 159888kb
input:
100 10000000
output:
451590067
result:
ok 1 number(s): "451590067"
Extra Test:
score: 0
Extra Test Passed