QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#177757#6417. Classical Summation ProblemduckierWA 7ms11392kbC++141015b2023-09-13 12:40:472023-09-13 12:40:47

Judging History

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

  • [2023-09-13 12:40:47]
  • 评测
  • 测评结果:WA
  • 用时:7ms
  • 内存:11392kb
  • [2023-09-13 12:40:47]
  • 提交

answer

#include <bits/stdc++.h>
#define int long long

using namespace std;

const int M = 998244353;
int n, k, fact[1000001];

int pow2(int a, int b) {
    if(b==0) return 1;
    int c = pow2(a, b/2);
    c = (c*c)%M;
    if(b%2==1) c = (c*a)%M;
    return c;
}

int inv(int a) {
    return pow2(a, M-2);
}

int choose(int a, int b) {
    return (((fact[a] * inv(fact[b]))%M) * inv(fact[a-b]))%M;
}

int mod(int a) {
    return (a%M + M)%M;
}

signed main() {
    fact[0] = 1;
    for(int i = 1; i <= 1000000; i++) fact[i] = (fact[i-1] * i)%M;
    cin>>n>>k;
    if(k%2==1) {
        int ans = pow2(n, k);
        ans = ((ans * (n+1))%M * inv(2))%M;
        cout << ans << '\n';
        return 0;
    }
    int ans =0;
    for(int i= 1; i <= n; i++) {
        ans += (((i * choose(k, k/2))%M * (mod(pow2(i, k/2) - pow2(i-1, k/2))))%M * pow2(n - i, k/2))%M;
    }
    ans %= M;
    k--;
    ans += ((pow2(n, k) * (n+1))%M * inv(2))%M;
    ans %= M;
    cout << ans << '\n';
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 7ms
memory: 11168kb

input:

3 2

output:

14

result:

ok 1 number(s): "14"

Test #2:

score: 0
Accepted
time: 7ms
memory: 11252kb

input:

5 3

output:

375

result:

ok 1 number(s): "375"

Test #3:

score: 0
Accepted
time: 7ms
memory: 11392kb

input:

2 2

output:

5

result:

ok 1 number(s): "5"

Test #4:

score: 0
Accepted
time: 3ms
memory: 11232kb

input:

10 9

output:

508778235

result:

ok 1 number(s): "508778235"

Test #5:

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

input:

69 3

output:

11497815

result:

ok 1 number(s): "11497815"

Test #6:

score: 0
Accepted
time: 7ms
memory: 11116kb

input:

994 515

output:

33689623

result:

ok 1 number(s): "33689623"

Test #7:

score: -100
Wrong Answer
time: 4ms
memory: 11232kb

input:

4476 6182

output:

216647822

result:

wrong answer 1st numbers differ - expected: '114894183', found: '216647822'