QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#619921#8339. Rooted Treekindow#AC ✓1162ms159884kbC++141.3kb2024-10-07 15:57:332024-10-07 15:57:33

Judging History

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

  • [2024-10-07 15:57:33]
  • 评测
  • 测评结果:AC
  • 用时:1162ms
  • 内存:159884kb
  • [2024-10-07 15:57:33]
  • 提交

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