QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#74339 | #4812. Counting Sequence | 12345678 | ML | 4813ms | 948452kb | C++14 | 2.0kb | 2023-01-31 19:54:50 | 2023-01-31 19:54:51 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
//#define int long long
#define mp make_pair
#define inf 1e9
#define pii pair <int, int>
const int mod = 998244353;
inline int read () {
int x = 0, f = 1;
char ch = getchar ();
while (ch < '0' || ch > '9') f = ((ch == '-') ? -1 : f), ch = getchar ();
while (ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar ();
return x * f;
}
inline void write (int x) {
if (x < 0) x = -x, putchar ('-');
if (x >= 10) write (x / 10);
putchar (x % 10 + '0');
}
inline int quickmod (int x, int y) {
int Ans = 1;
while (y) {
if (y & 1) Ans = (1ll * Ans * x) % mod;
x = (1ll * x * x) % mod;
y >>= 1;
}
return Ans;
}
inline void Add(int &x, int y) {
x += y;
if(x >= mod) x -= mod;
}
inline int add(int x, int y) {
x += y;
if(x >= mod) x -= mod;
return x;
}
const int B = 800;
int n, c;
namespace Sub1 {
int f[300005][2*B+5];
int solve() {
int Ans = 0;
for(int i = 1; i <= B; i++) f[i][i] = 1;
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= 2 * B; j++) {
if(j != 1 && i + j - 1 <= n) Add(f[i+j-1][j-1], 1ll * f[i][j] * c % mod);
if(i + j + 1 <= n) Add(f[i+j+1][j+1], f[i][j]);
}
}
for(int i = 1; i <= 2 * B; i++) Add(Ans, f[n][i]);
return Ans;
}
};//fr <= B
namespace Sub2 {
int f[2][600005];
const int AD = 300000;
int solve() {
int Ans = (n > B);
f[0][AD] = 1;
for(int len = 1, fl = 0; len <= 1000; len++, fl ^= 1) {
for(int i = 0; i <= 2 * AD; i++) f[fl^1][i] = add((i >= len) ? f[fl][i-len] : 0, (i + len <= 2 * AD) ? 1ll * f[fl][i+len] * c % mod : 0);
for(int i = B + 1; i <= n; i++) {
int j = n - (len + 1) * i + AD;
if(j >= 0 && j <= 2 * AD) Add(Ans, f[fl^1][j]);
}
}
return Ans;
}
};//fr > B, len < n / B
signed main () {
// freopen (".in", "r", stdin);
// freopen (".out", "w", stdout);
n = read(), c = read();
write(add(Sub1 :: solve(), Sub2 :: solve())), putchar('\n');
return 0;
}
/*
300000 114514
*/
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 2335ms
memory: 13356kb
input:
5 3
output:
8
result:
ok 1 number(s): "8"
Test #2:
score: 0
Accepted
time: 1436ms
memory: 13484kb
input:
1 0
output:
1
result:
ok 1 number(s): "1"
Test #3:
score: 0
Accepted
time: 2302ms
memory: 20840kb
input:
2022 39
output:
273239559
result:
ok 1 number(s): "273239559"
Test #4:
score: 0
Accepted
time: 2231ms
memory: 14624kb
input:
1 998244352
output:
1
result:
ok 1 number(s): "1"
Test #5:
score: 0
Accepted
time: 2352ms
memory: 13988kb
input:
1 12345678
output:
1
result:
ok 1 number(s): "1"
Test #6:
score: 0
Accepted
time: 2367ms
memory: 13748kb
input:
20 998998
output:
643731701
result:
ok 1 number(s): "643731701"
Test #7:
score: 0
Accepted
time: 2330ms
memory: 14664kb
input:
23 123
output:
947753998
result:
ok 1 number(s): "947753998"
Test #8:
score: 0
Accepted
time: 2338ms
memory: 14700kb
input:
50 5555
output:
745339864
result:
ok 1 number(s): "745339864"
Test #9:
score: 0
Accepted
time: 2315ms
memory: 14220kb
input:
60 6666
output:
690992218
result:
ok 1 number(s): "690992218"
Test #10:
score: 0
Accepted
time: 2402ms
memory: 14788kb
input:
100 50
output:
169678588
result:
ok 1 number(s): "169678588"
Test #11:
score: 0
Accepted
time: 2290ms
memory: 13680kb
input:
500 88888
output:
216149701
result:
ok 1 number(s): "216149701"
Test #12:
score: 0
Accepted
time: 2294ms
memory: 14436kb
input:
1000 213456
output:
270989457
result:
ok 1 number(s): "270989457"
Test #13:
score: 0
Accepted
time: 2307ms
memory: 20564kb
input:
2000 119988
output:
756425375
result:
ok 1 number(s): "756425375"
Test #14:
score: 0
Accepted
time: 2327ms
memory: 26708kb
input:
3000 998244352
output:
71841227
result:
ok 1 number(s): "71841227"
Test #15:
score: 0
Accepted
time: 2418ms
memory: 26900kb
input:
3000 555555555
output:
79880116
result:
ok 1 number(s): "79880116"
Test #16:
score: 0
Accepted
time: 2445ms
memory: 35084kb
input:
4321 1234
output:
949603993
result:
ok 1 number(s): "949603993"
Test #17:
score: 0
Accepted
time: 1498ms
memory: 39320kb
input:
5000 0
output:
5
result:
ok 1 number(s): "5"
Test #18:
score: 0
Accepted
time: 2431ms
memory: 39440kb
input:
5000 88888777
output:
833064960
result:
ok 1 number(s): "833064960"
Test #19:
score: 0
Accepted
time: 2372ms
memory: 39516kb
input:
5000 35557777
output:
696388498
result:
ok 1 number(s): "696388498"
Test #20:
score: 0
Accepted
time: 2455ms
memory: 70712kb
input:
10000 123456
output:
434296902
result:
ok 1 number(s): "434296902"
Test #21:
score: 0
Accepted
time: 2622ms
memory: 133384kb
input:
20000 555555
output:
34806915
result:
ok 1 number(s): "34806915"
Test #22:
score: 0
Accepted
time: 2797ms
memory: 196108kb
input:
30000 777888999
output:
58443551
result:
ok 1 number(s): "58443551"
Test #23:
score: 0
Accepted
time: 3341ms
memory: 321368kb
input:
50000 2
output:
90102905
result:
ok 1 number(s): "90102905"
Test #24:
score: 0
Accepted
time: 3578ms
memory: 446892kb
input:
70000 77998866
output:
202638568
result:
ok 1 number(s): "202638568"
Test #25:
score: 0
Accepted
time: 4043ms
memory: 634976kb
input:
100000 998244352
output:
360520717
result:
ok 1 number(s): "360520717"
Test #26:
score: 0
Accepted
time: 3951ms
memory: 634956kb
input:
100000 555555555
output:
613886009
result:
ok 1 number(s): "613886009"
Test #27:
score: 0
Accepted
time: 4813ms
memory: 948452kb
input:
150000 233333
output:
381065878
result:
ok 1 number(s): "381065878"
Test #28:
score: 0
Accepted
time: 4798ms
memory: 948452kb
input:
150000 20050117
output:
269891864
result:
ok 1 number(s): "269891864"
Test #29:
score: -100
Memory Limit Exceeded
input:
200000 114514