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;
}