QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#326148 | #4812. Counting Sequence | Tokido_Saya | WA | 1514ms | 190340kb | C++14 | 2.8kb | 2024-02-12 13:21:22 | 2024-02-12 13:21:22 |
Judging History
answer
// Sea, Your Memory
#include<bits/stdc++.h>
#define LL long long
#define DB double
#define MOD 998244353
#define ls(x) x << 1
#define rs(x) x << 1 | 1
#define lowbit(x) x & (-x)
#define PII pair<int, int>
#define MP make_pair
#define VI vector<int>
#define VII vector<int>::iterator
#define all(x) x.begin(), x.end()
#define EB emplace_back
#define SI set<int>
#define SII set<int>::iterator
#define QI queue<int>
#define fi first
#define se second
using namespace std;
template<typename T> void chkmn(T &a, const T &b) { (a > b) && (a = b); }
template<typename T> void chkmx(T &a, const T &b) { (a < b) && (a = b); }
int inc(const int &a, const int &b) { return a + b >= MOD ? a + b - MOD : a + b; }
int dec(const int &a, const int &b) { return a - b < 0 ? a - b + MOD : a - b; }
int mul(const int &a, const int &b) { return 1LL * a * b % MOD; }
int sqr(const int &a) { return 1LL * a * a % MOD; }
void Inc(int &a, const int &b) { ((a += b) >= MOD) && (a -= MOD); }
void Dec(int &a, const int &b) { ((a -= b) < 0) && (a += MOD); }
void Mul(int &a, const int &b) { a = 1LL * a * b % MOD; }
void Sqr(int &a) { a = 1LL * a * a % MOD; }
int qwqmi(int x, int k = MOD - 2)
{
int res = 1;
while(k)
{
if(k & 1) Mul(res, x);
Sqr(x), k >>= 1;
}
return res;
}
template<typename T> void read(T &x)
{
x = 0;
int f = 1;
char ch = getchar();
while(!isdigit(ch))
{
if(ch == '-')
f = -1;
ch = getchar();
}
while(isdigit(ch))
{
x = (x << 1) + (x << 3) + (ch ^ 48);
ch = getchar();
}
x = x * f;
}
const int N = 3e5 + 5;
const int B = 1933;
const int B2 = 2 * B + 5;
const int _B = N / B + 5;
int n, C;
int f[B2][B2];
int Solve1()
{
for(int i = 1; i < B; ++i)
f[i][i] = 1;
for(int s = 1; s <= n; ++s)
{
for(int x = 1; x <= 2 * B && s - x > 0; ++x)
{
int t = s - x, lst = t % B2, now = s % B2;
f[now][x] = inc(mul(C, f[lst][x + 1]), f[lst][x - 1]);
}
}
int res = 0;
for(int x = 1; x <= 2 * B; ++x)
Inc(res, f[n % B2][x]);
return res;
}
int g[2][_B * _B * 2][_B * 2];
int Solve2()
{
int now = 0, lst = 1;
g[0][_B * _B][_B] = 1;
int res = (n >= B);
for(int i = 2; i <= _B; ++i)
{
swap(now, lst);
memset(g, 0, sizeof(g));
for(int j = -i * (i - 1) / 2; j <= i * (i - 1) / 2; ++j)
for(int k = 1 - i; k <= i - 1; ++k)
{
j += _B * _B;
if(!g[lst][j][k + _B]) break;
Inc(g[now][j + k + 1][k + 1 + _B], g[lst][j][k + _B]);
Inc(g[now][j + k - 1][k - 1 + _B], g[lst][j][k + _B]);
}
for(int j = -i * (i + 1) / 2; j <= i * (i + 1) / 2; ++j)
if((n - j) % i == 0) for(int k = -i; k <= i; ++k)
Inc(res, g[now][j + _B * _B][k + _B]);
}
return res;
}
int main()
{
read(n), read(C);
int ans = inc(Solve1(), Solve2());
printf("%d\n", ans);
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1514ms
memory: 162812kb
input:
5 3
output:
8
result:
ok 1 number(s): "8"
Test #2:
score: 0
Accepted
time: 1496ms
memory: 162136kb
input:
1 0
output:
1
result:
ok 1 number(s): "1"
Test #3:
score: 0
Accepted
time: 1497ms
memory: 162912kb
input:
2022 39
output:
273239559
result:
ok 1 number(s): "273239559"
Test #4:
score: 0
Accepted
time: 1489ms
memory: 161104kb
input:
1 998244352
output:
1
result:
ok 1 number(s): "1"
Test #5:
score: 0
Accepted
time: 1476ms
memory: 162648kb
input:
1 12345678
output:
1
result:
ok 1 number(s): "1"
Test #6:
score: 0
Accepted
time: 1485ms
memory: 162380kb
input:
20 998998
output:
643731701
result:
ok 1 number(s): "643731701"
Test #7:
score: 0
Accepted
time: 1487ms
memory: 161316kb
input:
23 123
output:
947753998
result:
ok 1 number(s): "947753998"
Test #8:
score: 0
Accepted
time: 1488ms
memory: 162484kb
input:
50 5555
output:
745339864
result:
ok 1 number(s): "745339864"
Test #9:
score: 0
Accepted
time: 1480ms
memory: 162580kb
input:
60 6666
output:
690992218
result:
ok 1 number(s): "690992218"
Test #10:
score: 0
Accepted
time: 1475ms
memory: 162684kb
input:
100 50
output:
169678588
result:
ok 1 number(s): "169678588"
Test #11:
score: 0
Accepted
time: 1479ms
memory: 162676kb
input:
500 88888
output:
216149701
result:
ok 1 number(s): "216149701"
Test #12:
score: 0
Accepted
time: 1487ms
memory: 161416kb
input:
1000 213456
output:
270989457
result:
ok 1 number(s): "270989457"
Test #13:
score: 0
Accepted
time: 1481ms
memory: 162964kb
input:
2000 119988
output:
756425375
result:
ok 1 number(s): "756425375"
Test #14:
score: 0
Accepted
time: 1486ms
memory: 178764kb
input:
3000 998244352
output:
71841227
result:
ok 1 number(s): "71841227"
Test #15:
score: 0
Accepted
time: 1485ms
memory: 178552kb
input:
3000 555555555
output:
79880116
result:
ok 1 number(s): "79880116"
Test #16:
score: -100
Wrong Answer
time: 1502ms
memory: 190340kb
input:
4321 1234
output:
949602758
result:
wrong answer 1st numbers differ - expected: '949603993', found: '949602758'