BZOJ3780 : 数字统计
从低位到高位数位DP,f[i][j][k]表示已经填了后i位,转化的数字为j,后i位与x后i位的大小关系为k的方案数。
#include<cstdio> const int N=202,B=7,P=(1<<30)-1; char s[N];int T,n,y,i,j,k,t,a[N]; struct Num{ int x[B]; Num(){for(int i=0;i<B;i++)x[i]=0;} Num operator+(Num b){ Num c; for(int i=0;i<B;i++)c.x[i]=x[i]+b.x[i]; for(int i=0;i<B-1;i++)if(c.x[i]>P)c.x[i+1]++,c.x[i]&=P; return c; } Num operator-(Num b){ Num c; for(int i=0;i<B;i++)c.x[i]=x[i]-b.x[i]; for(int i=0;i<B-1;i++)if(c.x[i]<0)c.x[i+1]--,c.x[i]+=P+1; return c; } void operator+=(Num b){*this=*this+b;} void operator-=(Num b){*this=*this-b;} void write(){ int i; for(i=N-1;~i;i--)if(x[i/30]&(1<<i%30))break; if(i<0)putchar('0'); for(;~i;i--)putchar(x[i/30]&(1<<i%30)?'1':'0'); } }f[N][2][2],tmp,ans,one; Num cal(){ scanf("%s",s+1); for(i=1;i<=n;i++)a[n-i+1]=s[i]-'0',f[i][0][0]=f[i][0][1]=f[i][1][0]=f[i][1][1]=Num(); for(t=0;t<=1;t++)f[1][t][t<a[1]]+=one; for(i=1;i<n;i++)for(j=0;j<=1;j++)for(k=0;k<=1;k++)for(t=0;t<=1;t++)f[i+1][!j&&!t][t==a[i+1]?k:t<a[i+1]]+=f[i][j][k]; return f[n][y][1]; } int main(){ one.x[0]=1; scanf("%d",&T); while(T--){ scanf("%d%d",&n,&y); tmp=cal(),ans=cal(); for(k=a[1],i=2;i<=n;i++)k=!k&&!a[i]; if(k==y)ans+=one; ans-=tmp; ans.write();puts(""); } return 0; }