QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#72014 | #4812. Counting Sequence | wyzwyz | WA | 1954ms | 18840kb | C++14 | 1.5kb | 2023-01-13 17:43:26 | 2023-01-13 17:43:30 |
Judging History
answer
#include<cstdio>
#include<cctype>
#define maxn 505505
const int mod=998244353;
template<class T>
inline T read(){
T r=0,f=0;
char c;
while(!isdigit(c=getchar()))f|=(c=='-');
while(isdigit(c))r=(r<<1)+(r<<3)+(c^48),c=getchar();
return f?-r:r;
}
template<class T>
inline T min(T a,T b){
return a<b?a:b;
}
int n,w,B,lim;
int tmp[2222][2222],*f[maxn],g[2][maxn];
inline void Add(int &x,int y){
x+=y-mod,x+=(x>>31)&mod;
}
int main(){
n=read<int>();
w=read<int>()%mod;
while(B*(B+1)/2<n)B++;
for(lim=B+1;lim*(lim+1)/2-B*(B+1)/2<=n;lim++);
for(int i=0;i<=n;i++)f[i]=tmp[i%lim];
for(int i=1;i<=B;i++)f[i][i]=1;
for(int i=1;i<n;i++){
int Lim=min(min(i,lim),n-i+1);
for(int j=1;j<=Lim;j++){
if(!f[i][j])continue;
if(j>1)f[i+j-1][j-1]=(f[i+j-1][j-1]+1ll*f[i][j]*w)%mod;
if(i+j+1<=n)Add(f[i+j+1][j+1],f[i][j]);
f[i][j]=0;
}
}
int LIM=2*n/(B+1)-1,L=(LIM+1)*(B+1);
for(int i=LIM,o=LIM&1;~i;i--,o^=1){
int Lim=2*n/(i+1);
for(int j=B+1;j<=Lim;j++)
g[o][j*(i+1)]++;
if(!i)break;
L=min(L,(B+1)*(i+1));
for(int j=L;j<=2*n;j++){
if(!g[o][j])continue;
if(j+i<=2*n)Add(g[o^1][j+i],g[o][j]);
g[o^1][j-i]=(g[o^1][j-i]+1ll*g[o][j]*w)%mod;
g[o][j]=0;
}
L-=i;
}
long long ans=g[0][n];
for(int i=1;i<=lim;i++)ans+=f[n][i];
printf("%lld\n",ans%mod);
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 3ms
memory: 7608kb
input:
5 3
output:
8
result:
ok 1 number(s): "8"
Test #2:
score: 0
Accepted
time: 0ms
memory: 5572kb
input:
1 0
output:
1
result:
ok 1 number(s): "1"
Test #3:
score: 0
Accepted
time: 1ms
memory: 7676kb
input:
2022 39
output:
273239559
result:
ok 1 number(s): "273239559"
Test #4:
score: 0
Accepted
time: 0ms
memory: 5528kb
input:
1 998244352
output:
1
result:
ok 1 number(s): "1"
Test #5:
score: 0
Accepted
time: 2ms
memory: 5532kb
input:
1 12345678
output:
1
result:
ok 1 number(s): "1"
Test #6:
score: 0
Accepted
time: 2ms
memory: 5608kb
input:
20 998998
output:
643731701
result:
ok 1 number(s): "643731701"
Test #7:
score: 0
Accepted
time: 2ms
memory: 5580kb
input:
23 123
output:
947753998
result:
ok 1 number(s): "947753998"
Test #8:
score: 0
Accepted
time: 1ms
memory: 5588kb
input:
50 5555
output:
745339864
result:
ok 1 number(s): "745339864"
Test #9:
score: 0
Accepted
time: 2ms
memory: 7716kb
input:
60 6666
output:
690992218
result:
ok 1 number(s): "690992218"
Test #10:
score: 0
Accepted
time: 2ms
memory: 7640kb
input:
100 50
output:
169678588
result:
ok 1 number(s): "169678588"
Test #11:
score: 0
Accepted
time: 2ms
memory: 7576kb
input:
500 88888
output:
216149701
result:
ok 1 number(s): "216149701"
Test #12:
score: 0
Accepted
time: 1ms
memory: 5680kb
input:
1000 213456
output:
270989457
result:
ok 1 number(s): "270989457"
Test #13:
score: 0
Accepted
time: 3ms
memory: 7660kb
input:
2000 119988
output:
756425375
result:
ok 1 number(s): "756425375"
Test #14:
score: 0
Accepted
time: 3ms
memory: 9652kb
input:
3000 998244352
output:
71841227
result:
ok 1 number(s): "71841227"
Test #15:
score: 0
Accepted
time: 2ms
memory: 9716kb
input:
3000 555555555
output:
79880116
result:
ok 1 number(s): "79880116"
Test #16:
score: 0
Accepted
time: 5ms
memory: 7720kb
input:
4321 1234
output:
949603993
result:
ok 1 number(s): "949603993"
Test #17:
score: 0
Accepted
time: 5ms
memory: 7724kb
input:
5000 0
output:
5
result:
ok 1 number(s): "5"
Test #18:
score: 0
Accepted
time: 2ms
memory: 5680kb
input:
5000 88888777
output:
833064960
result:
ok 1 number(s): "833064960"
Test #19:
score: 0
Accepted
time: 5ms
memory: 5576kb
input:
5000 35557777
output:
696388498
result:
ok 1 number(s): "696388498"
Test #20:
score: 0
Accepted
time: 6ms
memory: 9844kb
input:
10000 123456
output:
434296902
result:
ok 1 number(s): "434296902"
Test #21:
score: 0
Accepted
time: 25ms
memory: 8028kb
input:
20000 555555
output:
34806915
result:
ok 1 number(s): "34806915"
Test #22:
score: 0
Accepted
time: 51ms
memory: 12088kb
input:
30000 777888999
output:
58443551
result:
ok 1 number(s): "58443551"
Test #23:
score: 0
Accepted
time: 111ms
memory: 8896kb
input:
50000 2
output:
90102905
result:
ok 1 number(s): "90102905"
Test #24:
score: 0
Accepted
time: 194ms
memory: 9424kb
input:
70000 77998866
output:
202638568
result:
ok 1 number(s): "202638568"
Test #25:
score: 0
Accepted
time: 366ms
memory: 13432kb
input:
100000 998244352
output:
360520717
result:
ok 1 number(s): "360520717"
Test #26:
score: 0
Accepted
time: 352ms
memory: 12540kb
input:
100000 555555555
output:
613886009
result:
ok 1 number(s): "613886009"
Test #27:
score: 0
Accepted
time: 684ms
memory: 15404kb
input:
150000 233333
output:
381065878
result:
ok 1 number(s): "381065878"
Test #28:
score: 0
Accepted
time: 667ms
memory: 13568kb
input:
150000 20050117
output:
269891864
result:
ok 1 number(s): "269891864"
Test #29:
score: 0
Accepted
time: 1059ms
memory: 16808kb
input:
200000 114514
output:
262861613
result:
ok 1 number(s): "262861613"
Test #30:
score: 0
Accepted
time: 1064ms
memory: 17036kb
input:
200000 1919810
output:
77872388
result:
ok 1 number(s): "77872388"
Test #31:
score: 0
Accepted
time: 1106ms
memory: 18840kb
input:
300000 0
output:
12
result:
ok 1 number(s): "12"
Test #32:
score: -100
Wrong Answer
time: 1954ms
memory: 18756kb
input:
300000 1
output:
153392993
result:
wrong answer 1st numbers differ - expected: '298803948', found: '153392993'