QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#359123#4812. Counting Sequencejames1BadCreeperWA 13ms14464kbC++141.2kb2024-03-20 13:23:522024-03-20 13:23:52

Judging History

This is the latest submission verdict.

  • [2024-03-20 13:23:52]
  • Judged
  • Verdict: WA
  • Time: 13ms
  • Memory: 14464kb
  • [2024-03-20 13:23:52]
  • Submitted

answer

#include <bits/stdc++.h>
using namespace std;
const int P = 998244353; 
const int N = 3e5 + 5, B = 800, T = 330; 

int n, c, ans; 
int f[(B + T) * 2][B + T], g[2][N * 2]; // sum = i, lst = j

int main(void) {
    ios::sync_with_stdio(0); 
    cin >> n >> c; 
    const int BB = (B + T) * 2; 
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= min(i, B + T - 1); ++j) {
            int I = i % BB; 
            if (j > 1) f[I][j] = (f[I][j] + f[(I - j + BB) % BB][j - 1]) % P; 
            if (j + 1 < B + T) f[I][j] = (f[I][j] + 1ll * f[(I - j + BB) % BB][j + 1] * c) % P; 
        }
        if (i < B) f[i][i] = 1; 
    }
    for (int i = 1; i <= min(n, B + T - 1); ++i) ans = (ans + f[n % BB][i]) % P; 

    if (n >= B) ++ans; 
    g[1][0 + N] = 1; 
    for (int i = 2, now = 0; i * B <= n * 2; ++i, now ^= 1)
        for (int j = max(-i * (i - 1) / 2, -i * B + 1); j <= min(i * (i - 1) / 2, n - i * B); ++j) {
            g[now][j + N] = (g[now ^ 1][j - (i - 1) + N] + 1ll * c * g[now ^ 1][j + (i - 1) + N]) % P; 
            if ((n - j) % i == 0) ans = (ans + g[now][j + N]) % P; 
        }
    cout << ans << "\n"; 
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

5 3

output:

8

result:

ok 1 number(s): "8"

Test #2:

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

input:

1 0

output:

1

result:

ok 1 number(s): "1"

Test #3:

score: 0
Accepted
time: 8ms
memory: 14464kb

input:

2022 39

output:

273239559

result:

ok 1 number(s): "273239559"

Test #4:

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

input:

1 998244352

output:

1

result:

ok 1 number(s): "1"

Test #5:

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

input:

1 12345678

output:

1

result:

ok 1 number(s): "1"

Test #6:

score: 0
Accepted
time: 1ms
memory: 5816kb

input:

20 998998

output:

643731701

result:

ok 1 number(s): "643731701"

Test #7:

score: 0
Accepted
time: 1ms
memory: 5728kb

input:

23 123

output:

947753998

result:

ok 1 number(s): "947753998"

Test #8:

score: 0
Accepted
time: 1ms
memory: 5908kb

input:

50 5555

output:

745339864

result:

ok 1 number(s): "745339864"

Test #9:

score: 0
Accepted
time: 1ms
memory: 5956kb

input:

60 6666

output:

690992218

result:

ok 1 number(s): "690992218"

Test #10:

score: 0
Accepted
time: 1ms
memory: 4024kb

input:

100 50

output:

169678588

result:

ok 1 number(s): "169678588"

Test #11:

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

input:

500 88888

output:

216149701

result:

ok 1 number(s): "216149701"

Test #12:

score: 0
Accepted
time: 6ms
memory: 10004kb

input:

1000 213456

output:

270989457

result:

ok 1 number(s): "270989457"

Test #13:

score: 0
Accepted
time: 8ms
memory: 14436kb

input:

2000 119988

output:

756425375

result:

ok 1 number(s): "756425375"

Test #14:

score: -100
Wrong Answer
time: 13ms
memory: 13580kb

input:

3000 998244352

output:

844534750

result:

wrong answer 1st numbers differ - expected: '71841227', found: '844534750'