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;

}