QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#333440 | #7754. Rolling For Days | lefy | TL | 336ms | 17596kb | C++14 | 2.9kb | 2024-02-19 22:08:06 | 2024-02-19 22:08:07 |
Judging History
answer
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e3+10,M=1<<12,mod=998244353;
int a[15],b[15],val[M],vb[M];
int c[N][N];
ll C(int x,int y){
if(x<0||y<0||x<y)return 0;
return c[x][y];
}
ll inv[N],Inv[N],sinv[N];
ll f[N][M],g[N][M],sf[N][M],sg[N][M],sg2[N][M],fac[N],G[N][M];
ll getmul(int l,int r){
if(l>r||r<=0)return 1;
if(!l)return fac[r];
return fac[r]*Inv[l-1]%mod;
}
ll getinv(int l,int r){
if(l>r||r<=0)return 1;
if(!l)return Inv[r];
return Inv[r]*fac[l-1]%mod;
}
ll getsum(int l,int r){
if(l>r||r<=0)return 0;
if(!l)return sinv[r];
return (sinv[r]-sinv[l-1]+mod)%mod;
}
int main() {
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)scanf("%d",&a[i]);
int st=0;
for(int i=1;i<=m;i++){
scanf("%d",&b[i]);
if(!b[i])st|=1<<i-1;
}
for(int s=0;s<(1<<m);s++)for(int i=1;i<=m;i++)if(s&(1<<i-1)){
val[s]+=a[i]-b[i],vb[s]+=b[i];
// cout<<i<<"\n";
}
for(int i=0;i<=n;i++){
c[i][0]=1;
for(int j=1;j<=i;j++)c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
}
sinv[0]=Inv[0]=Inv[1]=inv[1]=1;
sinv[1]=2;fac[1]=fac[0]=1;
for(int i=2;i<=n;i++){
fac[i]=fac[i-1]*i%mod;
sinv[i]=inv[i]=inv[mod%i]*(mod-mod/i)%mod,Inv[i]=Inv[i-1]*inv[i]%mod;
sinv[i]=(sinv[i]+sinv[i-1])%mod;
}
G[0][st]=sg2[0][st]=sf[0][st]=sg[0][st]=g[0][st]=1;
for(int i=1;i<=vb[(1<<m)-1];i++)for(int s=0;s<(1<<m);s++){
for(int x=1;x<=m;x++)if((s&(1<<x-1))&&b[x]){
for(int j=0;j<i;j++){
ll sum=C(i-1-vb[s-(1<<x-1)],b[x]-1)*getmul(a[x]-b[x]+1,a[x])%mod*getinv(n+1-i-val[s-(1<<x-1)],n-j-val[s-(1<<x-1)])%mod;
g[i][s]+=g[j][s-(1<<x-1)]*sum%mod;
f[i][s]+=f[j][s-(1<<x-1)]*sum%mod;
sum=sum*val[s-(1<<x-1)]%mod*g[j][s-(1<<x-1)]%mod*getsum(n+1-i-val[s-(1<<x-1)],n-j-val[s-(1<<x-1)])%mod;
f[i][s]+=sum;
f[i][s]%=mod,g[i][s]%=mod;
}
// ll sum=C(i-1-vb[s-(1<<x-1)],b[x]-1)*getmul(a[x]-b[x]+1,a[x])%mod;
// g[i][s]+=sg[i-1][s-(1<<x-1)]*sum%mod;
// f[i][s]+=sf[i-1][s-(1<<x-1)]*sum%mod;
// sum=sum*val[s-(1<<x-1)]%mod*sg2[i-1][s-(1<<x-1)]%mod;
// f[i][s]+=sum;
// g[i][s]%=mod,f[i][s]%=mod;
}
// sg[i][s]=(sg[i-1][s]+g[i][s])%mod;
// if(i<n)sg[i][s]=sg[i][s]*inv[n+1-i-1-val[s]]%mod;
// G[i][s]=(G[i-1][s]+g[i][s])%mod;
// sg2[i][s]=(sg2[i-1][s]+G[i][s]*inv[n+1-val[s]-i+1]%mod)%mod;
// sf[i][s]=sf[i-1][s];
// sf[i][s]=(sf[i][s]+f[i][s])%mod;
// if(i<n)sf[i][s]=sf[i][s]*inv[n+1-i-1-val[s]]%mod;
}
// cout<<f[vb[(1<<m)-1]][(1<<m)-1]<<"\n";
printf("%lld",(f[vb[(1<<m)-1]][(1<<m)-1]+vb[(1<<m)-1])%mod);
return 0;
}
/*20 12
1 2 1 1 2 4 1 3 2 1 1 1
1 0 0 1 0 0 1 0 2 0 1 1
691640771
*/
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3804kb
input:
2 2 1 1 1 1
output:
2
result:
ok answer is '2'
Test #2:
score: 0
Accepted
time: 1ms
memory: 3952kb
input:
4 2 2 2 2 1
output:
582309210
result:
ok answer is '582309210'
Test #3:
score: 0
Accepted
time: 0ms
memory: 3868kb
input:
5 5 1 1 1 1 1 0 0 0 0 1
output:
5
result:
ok answer is '5'
Test #4:
score: 0
Accepted
time: 1ms
memory: 3916kb
input:
4 4 1 1 1 1 1 1 1 0
output:
831870299
result:
ok answer is '831870299'
Test #5:
score: 0
Accepted
time: 0ms
memory: 3884kb
input:
5 2 4 1 2 1
output:
598946616
result:
ok answer is '598946616'
Test #6:
score: 0
Accepted
time: 1ms
memory: 3956kb
input:
5 2 3 2 3 1
output:
482484776
result:
ok answer is '482484776'
Test #7:
score: 0
Accepted
time: 1ms
memory: 3996kb
input:
5 5 1 1 1 1 1 0 1 1 1 0
output:
665496242
result:
ok answer is '665496242'
Test #8:
score: 0
Accepted
time: 0ms
memory: 4028kb
input:
3 3 1 1 1 1 1 0
output:
499122180
result:
ok answer is '499122180'
Test #9:
score: 0
Accepted
time: 1ms
memory: 4080kb
input:
5 5 1 1 1 1 1 1 0 1 1 1
output:
582309212
result:
ok answer is '582309212'
Test #10:
score: 0
Accepted
time: 0ms
memory: 3804kb
input:
3 2 2 1 2 0
output:
499122180
result:
ok answer is '499122180'
Test #11:
score: 0
Accepted
time: 1ms
memory: 4224kb
input:
20 5 1 6 7 2 4 0 1 3 1 4
output:
75028873
result:
ok answer is '75028873'
Test #12:
score: 0
Accepted
time: 1ms
memory: 4164kb
input:
15 5 4 2 3 4 2 2 1 1 2 1
output:
585494868
result:
ok answer is '585494868'
Test #13:
score: 0
Accepted
time: 1ms
memory: 3956kb
input:
20 4 5 4 3 8 1 2 2 3
output:
156108321
result:
ok answer is '156108321'
Test #14:
score: 0
Accepted
time: 1ms
memory: 4048kb
input:
15 2 6 9 2 8
output:
672033760
result:
ok answer is '672033760'
Test #15:
score: 0
Accepted
time: 6ms
memory: 4408kb
input:
20 12 1 2 1 1 2 4 1 3 2 1 1 1 1 0 0 1 0 0 1 0 2 0 1 1
output:
691640771
result:
ok answer is '691640771'
Test #16:
score: 0
Accepted
time: 16ms
memory: 4528kb
input:
19 12 1 1 1 2 1 2 2 1 2 4 1 1 1 1 0 1 1 0 1 1 0 2 1 0
output:
777326448
result:
ok answer is '777326448'
Test #17:
score: 0
Accepted
time: 1ms
memory: 4032kb
input:
20 2 19 1 1 1
output:
299473325
result:
ok answer is '299473325'
Test #18:
score: 0
Accepted
time: 1ms
memory: 4100kb
input:
19 2 14 5 10 1
output:
497380388
result:
ok answer is '497380388'
Test #19:
score: 0
Accepted
time: 1ms
memory: 4496kb
input:
100 5 10 25 6 19 40 0 2 4 5 11
output:
773338801
result:
ok answer is '773338801'
Test #20:
score: 0
Accepted
time: 0ms
memory: 4456kb
input:
64 5 1 12 13 33 5 1 0 1 20 0
output:
571823997
result:
ok answer is '571823997'
Test #21:
score: 0
Accepted
time: 1ms
memory: 4584kb
input:
100 4 15 38 24 23 0 20 0 1
output:
635309463
result:
ok answer is '635309463'
Test #22:
score: 0
Accepted
time: 0ms
memory: 5428kb
input:
88 5 15 25 9 19 20 8 15 9 18 17
output:
400310961
result:
ok answer is '400310961'
Test #23:
score: 0
Accepted
time: 194ms
memory: 6856kb
input:
100 12 2 2 13 9 13 7 2 1 6 15 17 13 0 0 5 7 10 7 0 1 0 0 4 4
output:
552732942
result:
ok answer is '552732942'
Test #24:
score: 0
Accepted
time: 219ms
memory: 6292kb
input:
59 12 7 6 3 5 4 6 5 2 5 6 5 5 4 5 2 5 3 6 0 2 1 0 3 3
output:
27023521
result:
ok answer is '27023521'
Test #25:
score: 0
Accepted
time: 2ms
memory: 5116kb
input:
100 3 10 60 30 0 28 21
output:
261595276
result:
ok answer is '261595276'
Test #26:
score: 0
Accepted
time: 1ms
memory: 4484kb
input:
84 2 39 45 4 23
output:
897695217
result:
ok answer is '897695217'
Test #27:
score: 0
Accepted
time: 336ms
memory: 17596kb
input:
1000 5 370 136 129 182 183 312 47 112 22 119
output:
705415872
result:
ok answer is '705415872'
Test #28:
score: 0
Accepted
time: 94ms
memory: 12764kb
input:
766 5 372 194 98 90 12 165 123 53 27 0
output:
870555094
result:
ok answer is '870555094'
Test #29:
score: 0
Accepted
time: 30ms
memory: 14084kb
input:
1000 2 374 626 175 591
output:
501708945
result:
ok answer is '501708945'
Test #30:
score: 0
Accepted
time: 0ms
memory: 10012kb
input:
701 1 701 413
output:
413
result:
ok answer is '413'
Test #31:
score: -100
Time Limit Exceeded
input:
1000 12 101 43 34 281 23 24 12 25 66 222 145 24 37 43 27 257 5 11 12 19 62 41 87 13