QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#359105 | #4812. Counting Sequence | james1BadCreeper | ML | 1486ms | 874092kb | C++14 | 1.0kb | 2024-03-20 12:54:09 | 2024-03-20 12:54:09 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
const int P = 998244353;
const int N = 3e5 + 5, B = 780, T = 330;
int n, c, ans;
int f[N][B + T], g[2][N * 2]; // sum = i, lst = j
int main(void) {
cin >> n >> c;
for (int i = 1; i <= min(n, B - 1); ++i) f[i][i] = 1;
for (int i = 2; i <= n; ++i) for (int j = 1; j <= min(i, B + T - 1); ++j) {
if (j > 1) f[i][j] = (f[i][j] + f[i - j][j - 1]) % P;
if (j + 1 < B + T) f[i][j] = (f[i][j] + 1ll * f[i - j][j + 1] * c) % P;
}
for (int i = 1; i <= min(n, B + T - 1); ++i) ans = (ans + f[n][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: 7588kb
input:
5 3
output:
8
result:
ok 1 number(s): "8"
Test #2:
score: 0
Accepted
time: 1ms
memory: 7640kb
input:
1 0
output:
1
result:
ok 1 number(s): "1"
Test #3:
score: 0
Accepted
time: 0ms
memory: 16596kb
input:
2022 39
output:
273239559
result:
ok 1 number(s): "273239559"
Test #4:
score: 0
Accepted
time: 1ms
memory: 7648kb
input:
1 998244352
output:
1
result:
ok 1 number(s): "1"
Test #5:
score: 0
Accepted
time: 1ms
memory: 7676kb
input:
1 12345678
output:
1
result:
ok 1 number(s): "1"
Test #6:
score: 0
Accepted
time: 1ms
memory: 7652kb
input:
20 998998
output:
643731701
result:
ok 1 number(s): "643731701"
Test #7:
score: 0
Accepted
time: 1ms
memory: 7792kb
input:
23 123
output:
947753998
result:
ok 1 number(s): "947753998"
Test #8:
score: 0
Accepted
time: 1ms
memory: 5628kb
input:
50 5555
output:
745339864
result:
ok 1 number(s): "745339864"
Test #9:
score: 0
Accepted
time: 1ms
memory: 7592kb
input:
60 6666
output:
690992218
result:
ok 1 number(s): "690992218"
Test #10:
score: 0
Accepted
time: 1ms
memory: 5716kb
input:
100 50
output:
169678588
result:
ok 1 number(s): "169678588"
Test #11:
score: 0
Accepted
time: 1ms
memory: 7676kb
input:
500 88888
output:
216149701
result:
ok 1 number(s): "216149701"
Test #12:
score: 0
Accepted
time: 2ms
memory: 9852kb
input:
1000 213456
output:
270989457
result:
ok 1 number(s): "270989457"
Test #13:
score: 0
Accepted
time: 9ms
memory: 16340kb
input:
2000 119988
output:
756425375
result:
ok 1 number(s): "756425375"
Test #14:
score: 0
Accepted
time: 7ms
memory: 20812kb
input:
3000 998244352
output:
71841227
result:
ok 1 number(s): "71841227"
Test #15:
score: 0
Accepted
time: 12ms
memory: 21192kb
input:
3000 555555555
output:
79880116
result:
ok 1 number(s): "79880116"
Test #16:
score: 0
Accepted
time: 27ms
memory: 26080kb
input:
4321 1234
output:
949603993
result:
ok 1 number(s): "949603993"
Test #17:
score: 0
Accepted
time: 34ms
memory: 29476kb
input:
5000 0
output:
5
result:
ok 1 number(s): "5"
Test #18:
score: 0
Accepted
time: 16ms
memory: 28760kb
input:
5000 88888777
output:
833064960
result:
ok 1 number(s): "833064960"
Test #19:
score: 0
Accepted
time: 22ms
memory: 26620kb
input:
5000 35557777
output:
696388498
result:
ok 1 number(s): "696388498"
Test #20:
score: 0
Accepted
time: 56ms
memory: 50884kb
input:
10000 123456
output:
434296902
result:
ok 1 number(s): "434296902"
Test #21:
score: 0
Accepted
time: 68ms
memory: 93828kb
input:
20000 555555
output:
34806915
result:
ok 1 number(s): "34806915"
Test #22:
score: 0
Accepted
time: 163ms
memory: 139460kb
input:
30000 777888999
output:
58443551
result:
ok 1 number(s): "58443551"
Test #23:
score: 0
Accepted
time: 372ms
memory: 224780kb
input:
50000 2
output:
90102905
result:
ok 1 number(s): "90102905"
Test #24:
score: 0
Accepted
time: 468ms
memory: 311284kb
input:
70000 77998866
output:
202638568
result:
ok 1 number(s): "202638568"
Test #25:
score: 0
Accepted
time: 716ms
memory: 439416kb
input:
100000 998244352
output:
360520717
result:
ok 1 number(s): "360520717"
Test #26:
score: 0
Accepted
time: 746ms
memory: 438712kb
input:
100000 555555555
output:
613886009
result:
ok 1 number(s): "613886009"
Test #27:
score: 0
Accepted
time: 1100ms
memory: 657296kb
input:
150000 233333
output:
381065878
result:
ok 1 number(s): "381065878"
Test #28:
score: 0
Accepted
time: 1063ms
memory: 658356kb
input:
150000 20050117
output:
269891864
result:
ok 1 number(s): "269891864"
Test #29:
score: 0
Accepted
time: 1486ms
memory: 874092kb
input:
200000 114514
output:
262861613
result:
ok 1 number(s): "262861613"
Test #30:
score: 0
Accepted
time: 1376ms
memory: 874016kb
input:
200000 1919810
output:
77872388
result:
ok 1 number(s): "77872388"
Test #31:
score: -100
Memory Limit Exceeded
input:
300000 0
output:
12