QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#333438#7754. Rolling For DayslefyWA 6ms14372kbC++142.9kb2024-02-19 22:06:322024-02-19 22:06:32

Judging History

你现在查看的是最新测评结果

  • [2024-02-19 22:06:32]
  • 评测
  • 测评结果:WA
  • 用时:6ms
  • 内存:14372kb
  • [2024-02-19 22:06:32]
  • 提交

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)]*g[j][s-(1<<x-1)]%mod*getsum(n+1-i-val[s-(1<<x-1)],n-j-val[s-(1<<x-1)]);
                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;
}
/*4 2
2 2
2 1
*/

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 2ms
memory: 14072kb

input:

2 2
1 1
1 1

output:

2

result:

ok answer is '2'

Test #2:

score: 0
Accepted
time: 2ms
memory: 14212kb

input:

4 2
2 2
2 1

output:

582309210

result:

ok answer is '582309210'

Test #3:

score: 0
Accepted
time: 2ms
memory: 14116kb

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: 0ms
memory: 14068kb

input:

4 4
1 1 1 1
1 1 1 0

output:

831870299

result:

ok answer is '831870299'

Test #5:

score: 0
Accepted
time: 1ms
memory: 14212kb

input:

5 2
4 1
2 1

output:

598946616

result:

ok answer is '598946616'

Test #6:

score: 0
Accepted
time: 1ms
memory: 14072kb

input:

5 2
3 2
3 1

output:

482484776

result:

ok answer is '482484776'

Test #7:

score: 0
Accepted
time: 0ms
memory: 14004kb

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: 13956kb

input:

3 3
1 1 1
1 1 0

output:

499122180

result:

ok answer is '499122180'

Test #9:

score: 0
Accepted
time: 2ms
memory: 14092kb

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: 13988kb

input:

3 2
2 1
2 0

output:

499122180

result:

ok answer is '499122180'

Test #11:

score: 0
Accepted
time: 0ms
memory: 14012kb

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: 0ms
memory: 14260kb

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: 2ms
memory: 14284kb

input:

20 4
5 4 3 8
1 2 2 3

output:

156108321

result:

ok answer is '156108321'

Test #14:

score: 0
Accepted
time: 0ms
memory: 14168kb

input:

15 2
6 9
2 8

output:

672033760

result:

ok answer is '672033760'

Test #15:

score: -100
Wrong Answer
time: 6ms
memory: 14372kb

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:

979367642

result:

wrong answer expected '691640771', found '979367642'