QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#340038 | #4812. Counting Sequence | tyl | AC ✓ | 2119ms | 943544kb | C++14 | 1.7kb | 2024-02-28 13:10:56 | 2024-02-28 13:10:56 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
const int N = 3e5 + 10, M = 800, mod = 998244353;
int n, c;
int dp[N][M], f[N], tmp[N], ans;
int inc(const int &a, const int &b) { return (a + b >= mod ? a + b - mod : a + b); }
int mul(const int &a, const int &b) { return 1ll * a * b % mod; }
void Inc(int &a, const int &b) { a = (a + b >= mod ? a + b - mod : a + b); }
int main() {
// freopen("aosora.in", "r", stdin);
// freopen("aosora.out", "w", stdout);
cin>>n>>c;
for(int i = 1; i <= min(n, M - 1); i++) dp[i][i] = 1;
for(int i = 2; i <= n; i++)
for(int j = 1; j <= min(i, M - 1); j++) {
if(j > 1) Inc(dp[i][j], dp[i - j][j - 1]);
if(j + 1 < M) Inc(dp[i][j], mul(dp[i - j][j + 1], c));
}
for(int i = 1; i < M; i++) Inc(ans, dp[n][i]);
tmp[M] = 1;
for(int i = M; i <= n; i++) Inc(tmp[i], dp[i - M][M - 1]);
// f[0][0] = 1;
// for(int i = 1; i < M; i++)
// for(int j = 0; j < N; j++) {
// f[i][j] = mul(f[i - 1][j], c);
// if(j >= i) Inc(f[i][j], f[i - 1][j - i]);
// }
f[0] = 1;
for(int k = 0; k < M; k++) {
if(k) {
for(int j = N - 1; j >= 0; j--) {
f[j] = mul(f[j], c);
if(j >= k) Inc(f[j], f[j - k]);
}
}
for(int i = M; i <= n; i++) {
for(int j = i; j <= n; j++) {
if(i > M && j > i) break;
int v = i == M ? tmp[j] : 1;
int res = n - j - i * k + k * (k + 1) / 2;
if(res >= 2 * N || res < 0 || (res & 1)) continue;
res >>= 1;
Inc(ans, mul(v, f[res]));
}
}
}
cout<<ans<<endl;
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 364ms
memory: 4888kb
input:
5 3
output:
8
result:
ok 1 number(s): "8"
Test #2:
score: 0
Accepted
time: 354ms
memory: 4836kb
input:
1 0
output:
1
result:
ok 1 number(s): "1"
Test #3:
score: 0
Accepted
time: 354ms
memory: 11192kb
input:
2022 39
output:
273239559
result:
ok 1 number(s): "273239559"
Test #4:
score: 0
Accepted
time: 372ms
memory: 4856kb
input:
1 998244352
output:
1
result:
ok 1 number(s): "1"
Test #5:
score: 0
Accepted
time: 351ms
memory: 4800kb
input:
1 12345678
output:
1
result:
ok 1 number(s): "1"
Test #6:
score: 0
Accepted
time: 354ms
memory: 4832kb
input:
20 998998
output:
643731701
result:
ok 1 number(s): "643731701"
Test #7:
score: 0
Accepted
time: 358ms
memory: 4780kb
input:
23 123
output:
947753998
result:
ok 1 number(s): "947753998"
Test #8:
score: 0
Accepted
time: 360ms
memory: 4944kb
input:
50 5555
output:
745339864
result:
ok 1 number(s): "745339864"
Test #9:
score: 0
Accepted
time: 356ms
memory: 4904kb
input:
60 6666
output:
690992218
result:
ok 1 number(s): "690992218"
Test #10:
score: 0
Accepted
time: 355ms
memory: 5116kb
input:
100 50
output:
169678588
result:
ok 1 number(s): "169678588"
Test #11:
score: 0
Accepted
time: 355ms
memory: 6396kb
input:
500 88888
output:
216149701
result:
ok 1 number(s): "216149701"
Test #12:
score: 0
Accepted
time: 356ms
memory: 7888kb
input:
1000 213456
output:
270989457
result:
ok 1 number(s): "270989457"
Test #13:
score: 0
Accepted
time: 366ms
memory: 10984kb
input:
2000 119988
output:
756425375
result:
ok 1 number(s): "756425375"
Test #14:
score: 0
Accepted
time: 369ms
memory: 14176kb
input:
3000 998244352
output:
71841227
result:
ok 1 number(s): "71841227"
Test #15:
score: 0
Accepted
time: 363ms
memory: 14196kb
input:
3000 555555555
output:
79880116
result:
ok 1 number(s): "79880116"
Test #16:
score: 0
Accepted
time: 374ms
memory: 18360kb
input:
4321 1234
output:
949603993
result:
ok 1 number(s): "949603993"
Test #17:
score: 0
Accepted
time: 380ms
memory: 20452kb
input:
5000 0
output:
5
result:
ok 1 number(s): "5"
Test #18:
score: 0
Accepted
time: 375ms
memory: 20308kb
input:
5000 88888777
output:
833064960
result:
ok 1 number(s): "833064960"
Test #19:
score: 0
Accepted
time: 376ms
memory: 20428kb
input:
5000 35557777
output:
696388498
result:
ok 1 number(s): "696388498"
Test #20:
score: 0
Accepted
time: 405ms
memory: 36040kb
input:
10000 123456
output:
434296902
result:
ok 1 number(s): "434296902"
Test #21:
score: 0
Accepted
time: 451ms
memory: 67280kb
input:
20000 555555
output:
34806915
result:
ok 1 number(s): "34806915"
Test #22:
score: 0
Accepted
time: 504ms
memory: 98744kb
input:
30000 777888999
output:
58443551
result:
ok 1 number(s): "58443551"
Test #23:
score: 0
Accepted
time: 640ms
memory: 161320kb
input:
50000 2
output:
90102905
result:
ok 1 number(s): "90102905"
Test #24:
score: 0
Accepted
time: 745ms
memory: 223816kb
input:
70000 77998866
output:
202638568
result:
ok 1 number(s): "202638568"
Test #25:
score: 0
Accepted
time: 905ms
memory: 317592kb
input:
100000 998244352
output:
360520717
result:
ok 1 number(s): "360520717"
Test #26:
score: 0
Accepted
time: 916ms
memory: 317596kb
input:
100000 555555555
output:
613886009
result:
ok 1 number(s): "613886009"
Test #27:
score: 0
Accepted
time: 1218ms
memory: 474168kb
input:
150000 233333
output:
381065878
result:
ok 1 number(s): "381065878"
Test #28:
score: 0
Accepted
time: 1194ms
memory: 474212kb
input:
150000 20050117
output:
269891864
result:
ok 1 number(s): "269891864"
Test #29:
score: 0
Accepted
time: 1418ms
memory: 630424kb
input:
200000 114514
output:
262861613
result:
ok 1 number(s): "262861613"
Test #30:
score: 0
Accepted
time: 1421ms
memory: 630484kb
input:
200000 1919810
output:
77872388
result:
ok 1 number(s): "77872388"
Test #31:
score: 0
Accepted
time: 2092ms
memory: 943328kb
input:
300000 0
output:
12
result:
ok 1 number(s): "12"
Test #32:
score: 0
Accepted
time: 2007ms
memory: 943452kb
input:
300000 1
output:
298803948
result:
ok 1 number(s): "298803948"
Test #33:
score: 0
Accepted
time: 1894ms
memory: 943384kb
input:
300000 2
output:
106751203
result:
ok 1 number(s): "106751203"
Test #34:
score: 0
Accepted
time: 2024ms
memory: 943544kb
input:
300000 1234
output:
427045479
result:
ok 1 number(s): "427045479"
Test #35:
score: 0
Accepted
time: 2081ms
memory: 943392kb
input:
300000 2345
output:
441593553
result:
ok 1 number(s): "441593553"
Test #36:
score: 0
Accepted
time: 2106ms
memory: 943432kb
input:
300000 20041115
output:
580367993
result:
ok 1 number(s): "580367993"
Test #37:
score: 0
Accepted
time: 2074ms
memory: 943456kb
input:
300000 20050117
output:
579859619
result:
ok 1 number(s): "579859619"
Test #38:
score: 0
Accepted
time: 2021ms
memory: 943356kb
input:
300000 22223333
output:
596066085
result:
ok 1 number(s): "596066085"
Test #39:
score: 0
Accepted
time: 2058ms
memory: 943368kb
input:
300000 175846372
output:
660364393
result:
ok 1 number(s): "660364393"
Test #40:
score: 0
Accepted
time: 2067ms
memory: 943328kb
input:
299999 9999999
output:
954865020
result:
ok 1 number(s): "954865020"
Test #41:
score: 0
Accepted
time: 2019ms
memory: 943372kb
input:
299998 55556666
output:
904862432
result:
ok 1 number(s): "904862432"
Test #42:
score: 0
Accepted
time: 1445ms
memory: 601532kb
input:
190733 31756136
output:
880544587
result:
ok 1 number(s): "880544587"
Test #43:
score: 0
Accepted
time: 1414ms
memory: 625916kb
input:
198497 463488298
output:
185220207
result:
ok 1 number(s): "185220207"
Test #44:
score: 0
Accepted
time: 2119ms
memory: 943380kb
input:
299997 0
output:
16
result:
ok 1 number(s): "16"
Test #45:
score: 0
Accepted
time: 1280ms
memory: 531020kb
input:
168168 296157813
output:
798716760
result:
ok 1 number(s): "798716760"
Test #46:
score: 0
Accepted
time: 973ms
memory: 363112kb
input:
114514 1919810
output:
783513290
result:
ok 1 number(s): "783513290"