QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#362283#8339. Rooted Treecomeintocalm#AC ✓280ms81548kbC++171.0kb2024-03-23 14:54:302024-03-23 14:54:31

Judging History

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

  • [2024-03-23 14:54:31]
  • 评测
  • 测评结果:AC
  • 用时:280ms
  • 内存:81548kb
  • [2024-03-23 14:54:30]
  • 提交

answer

#include <bits/stdc++.h>

const int mod = 1e9 + 9;

int qpow(int base, int b, int ret = 1) {
    for (; b; base = 1ll * base * base % mod, b >>= 1)
        if (b & 1)
            ret = 1ll * ret * base % mod;
    return ret;
}

int main() {
    int m, k;
    std::cin >> m >> k;
    std::vector<int> fac(k + 1), inv(k + 1);
    fac[0] = inv[0] = 1;
    for (int i = 1; i <= k; ++i) {
        fac[i] = 1ll * fac[i - 1] * (1ll * i * (m - 1) % mod + 1) % mod;
    }
    inv[k] = qpow(fac[k], mod - 2);
    for (int i = k - 1; i >= 1; --i) {
        inv[i] = 1ll * inv[i + 1] * (1ll * (i + 1) * (m - 1) % mod + 1) % mod;
    }
    for (int i = 1; i <= k; ++i) {
        inv[i] = 1ll * inv[i] * fac[i - 1] % mod;
    }
    int sum = 0, ans = 0;
    for (int i = 0; i < k; ++i) {
        int exp = 1ll * sum * inv[i] % mod;
        (sum += 1ll * (1ll * (exp + 1) * m % mod + mod - exp) % mod) %= mod;
        (ans += 1ll * (exp + 1) * m % mod) %= mod;
        // std::cout << sum << ' ' << inv[i] << std::endl;
    }
    std::cout << ans << std::endl;
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3656kb

input:

6 2

output:

18

result:

ok 1 number(s): "18"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3568kb

input:

2 6

output:

600000038

result:

ok 1 number(s): "600000038"

Test #3:

score: 0
Accepted
time: 14ms
memory: 8140kb

input:

83 613210

output:

424200026

result:

ok 1 number(s): "424200026"

Test #4:

score: 0
Accepted
time: 180ms
memory: 55724kb

input:

48 6713156

output:

198541581

result:

ok 1 number(s): "198541581"

Test #5:

score: 0
Accepted
time: 0ms
memory: 3560kb

input:

1 111

output:

6216

result:

ok 1 number(s): "6216"

Test #6:

score: 0
Accepted
time: 201ms
memory: 60240kb

input:

28 7304152

output:

457266679

result:

ok 1 number(s): "457266679"

Test #7:

score: 0
Accepted
time: 114ms
memory: 35236kb

input:

38 4101162

output:

232117382

result:

ok 1 number(s): "232117382"

Test #8:

score: 0
Accepted
time: 280ms
memory: 80752kb

input:

51 9921154

output:

340670552

result:

ok 1 number(s): "340670552"

Test #9:

score: 0
Accepted
time: 52ms
memory: 17460kb

input:

79 1801157

output:

620550406

result:

ok 1 number(s): "620550406"

Test #10:

score: 0
Accepted
time: 147ms
memory: 45596kb

input:

22 5417157

output:

457449071

result:

ok 1 number(s): "457449071"

Test #11:

score: 0
Accepted
time: 88ms
memory: 28320kb

input:

25 3210162

output:

36368303

result:

ok 1 number(s): "36368303"

Test #12:

score: 0
Accepted
time: 80ms
memory: 25968kb

input:

67 2919160

output:

935195555

result:

ok 1 number(s): "935195555"

Test #13:

score: 0
Accepted
time: 234ms
memory: 70368kb

input:

77 8613163

output:

482832472

result:

ok 1 number(s): "482832472"

Test #14:

score: 0
Accepted
time: 264ms
memory: 81324kb

input:

90 10000000

output:

275581651

result:

ok 1 number(s): "275581651"

Test #15:

score: 0
Accepted
time: 268ms
memory: 81548kb

input:

99 9999999

output:

126087169

result:

ok 1 number(s): "126087169"

Test #16:

score: 0
Accepted
time: 273ms
memory: 81188kb

input:

100 10000000

output:

451590067

result:

ok 1 number(s): "451590067"

Extra Test:

score: 0
Extra Test Passed