QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#619921 | #8339. Rooted Tree | kindow# | AC ✓ | 1162ms | 159884kb | C++14 | 1.3kb | 2024-10-07 15:57:33 | 2024-10-07 15:57:33 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int p = 1e9+9;
const int N = 1e7+5;
#define pb push_back
int m, k;
ll pm, A[N], g, f[N];
ll qpow(ll a, ll b)
{
ll ans = 1;
while(b)
{
if(b & 1)ans = (ans * a) % p;
a = (a * a) % p;
b >>= 1;
}
return ans;
}
// ll f(int x)
// {
// if(x == 0)return 0;
// x--;
// ll pmn;
// if(x > 0) pmn = (1LL * qpow(A[x], p-2) * A[x-1]) % p;
// else pmn = 1;
// ll ans = 1LL * qpow(m, 2) * pm % p + (1LL * (1 + 1LL * (m-1) * pmn % p) * f(x)) % p;
// return ans % p;
// }
int main()
{
ios::sync_with_stdio(0);
cin.tie(0); cout.tie(0);
cin >> m >> k;
if(m == 1)
{
ll ans = 1LL * k * (k + 1) / 2 % p;
cout << ans << '\n';
return 0;
}
pm = qpow(m-1, p-2);
A[0] = 1;
for(int i = 1; i <= k; ++i)
{
A[i] = ((1LL * A[i-1] * ((1 + 1LL * (m-1) * i % p) % p)) % p) % p;
}
f[0] = 0;
for(int i = 1; i <= k; ++i)
{
ll pmn;
int x = i-1;
if(x > 0) pmn = (1LL * qpow(A[x], p-2) * A[x-1]) % p;
else pmn = 1;
f[i] = 1LL * qpow(m, 2) * pm % p + (1LL * (1 + 1LL * (m-1) * pmn % p) * f[x]) % p;
f[i] = f[i] % p;
}
ll g = 1LL * ((1LL * k * m) % p) * pm % p;
ll ans = ((f[k] - g) % p + p) % p;
cout << ans << '\n';
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 5688kb
input:
6 2
output:
18
result:
ok 1 number(s): "18"
Test #2:
score: 0
Accepted
time: 0ms
memory: 5636kb
input:
2 6
output:
600000038
result:
ok 1 number(s): "600000038"
Test #3:
score: 0
Accepted
time: 74ms
memory: 18352kb
input:
83 613210
output:
424200026
result:
ok 1 number(s): "424200026"
Test #4:
score: 0
Accepted
time: 772ms
memory: 112692kb
input:
48 6713156
output:
198541581
result:
ok 1 number(s): "198541581"
Test #5:
score: 0
Accepted
time: 0ms
memory: 3640kb
input:
1 111
output:
6216
result:
ok 1 number(s): "6216"
Test #6:
score: 0
Accepted
time: 826ms
memory: 120632kb
input:
28 7304152
output:
457266679
result:
ok 1 number(s): "457266679"
Test #7:
score: 0
Accepted
time: 481ms
memory: 70728kb
input:
38 4101162
output:
232117382
result:
ok 1 number(s): "232117382"
Test #8:
score: 0
Accepted
time: 1140ms
memory: 159824kb
input:
51 9921154
output:
340670552
result:
ok 1 number(s): "340670552"
Test #9:
score: 0
Accepted
time: 198ms
memory: 35972kb
input:
79 1801157
output:
620550406
result:
ok 1 number(s): "620550406"
Test #10:
score: 0
Accepted
time: 620ms
memory: 92724kb
input:
22 5417157
output:
457449071
result:
ok 1 number(s): "457449071"
Test #11:
score: 0
Accepted
time: 375ms
memory: 56688kb
input:
25 3210162
output:
36368303
result:
ok 1 number(s): "36368303"
Test #12:
score: 0
Accepted
time: 333ms
memory: 51300kb
input:
67 2919160
output:
935195555
result:
ok 1 number(s): "935195555"
Test #13:
score: 0
Accepted
time: 989ms
memory: 141568kb
input:
77 8613163
output:
482832472
result:
ok 1 number(s): "482832472"
Test #14:
score: 0
Accepted
time: 1162ms
memory: 159872kb
input:
90 10000000
output:
275581651
result:
ok 1 number(s): "275581651"
Test #15:
score: 0
Accepted
time: 1158ms
memory: 159884kb
input:
99 9999999
output:
126087169
result:
ok 1 number(s): "126087169"
Test #16:
score: 0
Accepted
time: 1145ms
memory: 159828kb
input:
100 10000000
output:
451590067
result:
ok 1 number(s): "451590067"
Extra Test:
score: 0
Extra Test Passed