QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#779943 | #8339. Rooted Tree | Zeoy | AC ✓ | 238ms | 198892kb | C++20 | 1.6kb | 2024-11-24 22:55:56 | 2024-11-24 22:56:32 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define endl "\n"
using ll = long long;
const int mod = 1e9 + 9;
int add(int a, int b) {
int c = a + b;
if (c >= mod) c -= mod;
return c;
}
int sub(int a, int b) {
int c = a - b;
if (c < 0) c += mod;
return c;
}
int mul(int a, int b) {
ll c = 1ll * a * b;
if (c >= mod) c %= mod;
return c;
}
inline int ksm(int a, int b) {
int res = 1;
for (; b; b >>= 1) {
if (b & 1) {
res = 1ll * res * a % mod;
}
a = 1ll * a * a % mod;
}
return res;
}
const int N = 10000010;
int f[N], cnt[N], inv[N], prod[N], iprod[N];
void solve() {
int M, K; cin >> M >> K;
cnt[0] = 1;
for (int i = 1; i <= K; ++i) cnt[i] = cnt[i - 1] + M - 1;
prod[0] = 1;
for (int i = 1; i <= K; ++i) prod[i] = mul(prod[i - 1], cnt[i]);
iprod[K] = ksm(prod[K], mod - 2);
for (int i = K - 1; i >= 1; --i) iprod[i] = mul(iprod[i + 1], cnt[i + 1]);
for (int i = 1; i <= K; ++i) inv[i] = mul(prod[i - 1], iprod[i]);
int ans = 0;
f[0] = 0;
for (int i = 1; i <= K; ++i) {
cnt[i] = cnt[i - 1] + M - 1;
int res = mul(M, add(mul(f[i - 1], inv[i - 1]), 1));
ans = add(ans, res);
f[i] = 0;
f[i] = add(f[i], res);
f[i] = add(f[i], f[i - 1]);
f[i] = sub(f[i], mul(f[i - 1], inv[i - 1]));
}
cout << ans << endl;
}
signed main() {
ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
int T = 1;
// cin >> T;
while (T--) {
solve();
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 2ms
memory: 11820kb
input:
6 2
output:
18
result:
ok 1 number(s): "18"
Test #2:
score: 0
Accepted
time: 1ms
memory: 11824kb
input:
2 6
output:
600000038
result:
ok 1 number(s): "600000038"
Test #3:
score: 0
Accepted
time: 7ms
memory: 28468kb
input:
83 613210
output:
424200026
result:
ok 1 number(s): "424200026"
Test #4:
score: 0
Accepted
time: 154ms
memory: 138704kb
input:
48 6713156
output:
198541581
result:
ok 1 number(s): "198541581"
Test #5:
score: 0
Accepted
time: 2ms
memory: 11760kb
input:
1 111
output:
6216
result:
ok 1 number(s): "6216"
Test #6:
score: 0
Accepted
time: 151ms
memory: 149728kb
input:
28 7304152
output:
457266679
result:
ok 1 number(s): "457266679"
Test #7:
score: 0
Accepted
time: 98ms
memory: 91212kb
input:
38 4101162
output:
232117382
result:
ok 1 number(s): "232117382"
Test #8:
score: 0
Accepted
time: 234ms
memory: 198336kb
input:
51 9921154
output:
340670552
result:
ok 1 number(s): "340670552"
Test #9:
score: 0
Accepted
time: 36ms
memory: 53472kb
input:
79 1801157
output:
620550406
result:
ok 1 number(s): "620550406"
Test #10:
score: 0
Accepted
time: 131ms
memory: 112748kb
input:
22 5417157
output:
457449071
result:
ok 1 number(s): "457449071"
Test #11:
score: 0
Accepted
time: 60ms
memory: 76188kb
input:
25 3210162
output:
36368303
result:
ok 1 number(s): "36368303"
Test #12:
score: 0
Accepted
time: 62ms
memory: 69760kb
input:
67 2919160
output:
935195555
result:
ok 1 number(s): "935195555"
Test #13:
score: 0
Accepted
time: 204ms
memory: 173284kb
input:
77 8613163
output:
482832472
result:
ok 1 number(s): "482832472"
Test #14:
score: 0
Accepted
time: 238ms
memory: 198892kb
input:
90 10000000
output:
275581651
result:
ok 1 number(s): "275581651"
Test #15:
score: 0
Accepted
time: 222ms
memory: 198880kb
input:
99 9999999
output:
126087169
result:
ok 1 number(s): "126087169"
Test #16:
score: 0
Accepted
time: 219ms
memory: 198892kb
input:
100 10000000
output:
451590067
result:
ok 1 number(s): "451590067"
Extra Test:
score: 0
Extra Test Passed