QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#83916 | #4812. Counting Sequence | Scintilla | WA | 224ms | 14840kb | C++14 | 2.6kb | 2023-03-04 10:43:56 | 2023-03-04 10:43:58 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define rep(i, s, e) for (int i = s; i <= e; ++i)
#define drep(i, s, e) for (int i = s; i >= e; --i)
#define file(a) freopen(#a".in", "r", stdin), freopen(#a".out", "w", stdout)
#define pv(a) cout << #a << " = " << a << endl
#define pa(a, l, r) cout << #a " : "; rep(_, l, r) cout << a[_] << ' '; cout << endl
const int P = 998244353;
const int N = 3e5 + 10;
const int B = 800;
int read() {
int x = 0, f = 1; char c = getchar();
for (; c < '0' || c > '9'; c = getchar()) if (c == '-') f = -1;
for (; c >= '0' && c <= '9'; c = getchar()) x = x * 10 + c - 48;
return x * f;
}
int inc(int a, int b) { return (a += b) >= P ? a - P : a; }
int dec(int a, int b) { return (a -= b) < 0 ? a + P : a; }
int mul(int a, int b) { return 1ll * a * b % P; }
void add(int &a, int b) { (a += b) >= P ? a -= P : 1; }
void sub(int &a, int b) { (a -= b) < 0 ? a += P : 1; }
int qpow(int a, int b) { int res = 1; for (; b; b >>= 1, a = mul(a, a)) if (b & 1) res = mul(res, a); return res; }
int n, c, ans;
namespace small {
const int M = 1.3e3 + 10;
const int L = 1.3e3;
int f[M][M];
void Main() {
for (int i = 1, o = 1; i <= n; ++ i, o = (o + 1) % M) {
fill(f[o] + 1, f[o] + L + 1, 0);
if (i <= B) f[o][i] = 1;
// cout << "i, o = " << i << ' ' << o << endl;
rep(j, 1, L) {
add(f[o][j], f[(o - j + M) % M][j - 1]);
add(f[o][j], mul(f[(o - j + M) % M][j + 1], c));
}
}
rep(i, 1, B) add(ans, f[n % M][i]);
}
}
namespace large {
int pool[200000000], cur, *f[N], *g, c2[N], lim[N];
void Main() {
rep(i, 1, B) c2[i] = c2[i - 1] + i - 1;
rep(i, B + 1, n) {
for (lim[i] = 1; i * lim[i] - c2[lim[i]] <= n; ++ lim[i]);
cur += c2[lim[i]];
f[i] = pool + cur;
cur += c2[lim[i]];
}
// pa(c2, 1, B);
// pa(lim, B + 1, n);
cur += c2[B];
g = pool + cur;
cur += c2[B];
if (B < n) {
rep(i, B + 1, n) f[i][0] = 1;
add(ans, 1);
}
rep(i, 2, n) {
rep(s, -c2[i], c2[i]) g[s] = 0;
for (int j = B + 1; j <= n && i < lim[j]; ++ j) {
rep(s, -c2[i], c2[i]) g[s] = f[j][s];
// cout << "i, j = " << i << ' ' << j << endl;
rep(s, -c2[i], c2[i]) {
f[j][s] = g[s - i + 1];
add(f[j][s], mul(f[j + 1][s + i - 1], c));
if (i * j + s == n) {
// cout << "i, j = " << i << ' ' << j << endl;
add(ans, f[j][s]);
}
}
}
}
}
}
int main() {
n = read(), c = read();
small::Main();
large::Main();
printf("%d\n", ans);
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 3ms
memory: 3820kb
input:
5 3
output:
8
result:
ok 1 number(s): "8"
Test #2:
score: 0
Accepted
time: 3ms
memory: 3520kb
input:
1 0
output:
1
result:
ok 1 number(s): "1"
Test #3:
score: 0
Accepted
time: 50ms
memory: 12812kb
input:
2022 39
output:
273239559
result:
ok 1 number(s): "273239559"
Test #4:
score: 0
Accepted
time: 0ms
memory: 3508kb
input:
1 998244352
output:
1
result:
ok 1 number(s): "1"
Test #5:
score: 0
Accepted
time: 2ms
memory: 3788kb
input:
1 12345678
output:
1
result:
ok 1 number(s): "1"
Test #6:
score: 0
Accepted
time: 3ms
memory: 3832kb
input:
20 998998
output:
643731701
result:
ok 1 number(s): "643731701"
Test #7:
score: 0
Accepted
time: 3ms
memory: 3668kb
input:
23 123
output:
947753998
result:
ok 1 number(s): "947753998"
Test #8:
score: 0
Accepted
time: 4ms
memory: 3720kb
input:
50 5555
output:
745339864
result:
ok 1 number(s): "745339864"
Test #9:
score: 0
Accepted
time: 0ms
memory: 3816kb
input:
60 6666
output:
690992218
result:
ok 1 number(s): "690992218"
Test #10:
score: 0
Accepted
time: 2ms
memory: 4048kb
input:
100 50
output:
169678588
result:
ok 1 number(s): "169678588"
Test #11:
score: 0
Accepted
time: 12ms
memory: 7244kb
input:
500 88888
output:
216149701
result:
ok 1 number(s): "216149701"
Test #12:
score: 0
Accepted
time: 29ms
memory: 11164kb
input:
1000 213456
output:
270989457
result:
ok 1 number(s): "270989457"
Test #13:
score: 0
Accepted
time: 44ms
memory: 12744kb
input:
2000 119988
output:
756425375
result:
ok 1 number(s): "756425375"
Test #14:
score: 0
Accepted
time: 64ms
memory: 12968kb
input:
3000 998244352
output:
71841227
result:
ok 1 number(s): "71841227"
Test #15:
score: 0
Accepted
time: 46ms
memory: 12816kb
input:
3000 555555555
output:
79880116
result:
ok 1 number(s): "79880116"
Test #16:
score: 0
Accepted
time: 66ms
memory: 12904kb
input:
4321 1234
output:
949603993
result:
ok 1 number(s): "949603993"
Test #17:
score: 0
Accepted
time: 79ms
memory: 12920kb
input:
5000 0
output:
5
result:
ok 1 number(s): "5"
Test #18:
score: 0
Accepted
time: 79ms
memory: 12800kb
input:
5000 88888777
output:
833064960
result:
ok 1 number(s): "833064960"
Test #19:
score: 0
Accepted
time: 74ms
memory: 12872kb
input:
5000 35557777
output:
696388498
result:
ok 1 number(s): "696388498"
Test #20:
score: 0
Accepted
time: 122ms
memory: 13328kb
input:
10000 123456
output:
434296902
result:
ok 1 number(s): "434296902"
Test #21:
score: -100
Wrong Answer
time: 224ms
memory: 14840kb
input:
20000 555555
output:
80208541
result:
wrong answer 1st numbers differ - expected: '34806915', found: '80208541'