QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#779943#8339. Rooted TreeZeoyAC ✓238ms198892kbC++201.6kb2024-11-24 22:55:562024-11-24 22:56:32

Judging History

你现在查看的是最新测评结果

  • [2024-11-24 22:56:32]
  • 评测
  • 测评结果:AC
  • 用时:238ms
  • 内存:198892kb
  • [2024-11-24 22:55:56]
  • 提交

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