BZOJ3329 : Xorequ

第一问：

打表可得规律：当且仅当x&(x<<1)=0时才会是解，于是数位DP

f[i][j][k]表示二进制中前i位，上一位是j，前i位是否等于n的方案数

第二问：

打表可得规律：答案为斐波那契数列第n+2项，矩阵快速幂即可

 

#include<cstdio>

typedef long long ll;

struct mat{

  ll a[2][2];

  mat(){a[0][0]=a[0][1]=a[1][0]=a[1][1]=0;}

  mat operator*(mat b){

    mat c;

    for(int i=0,j,k;i<2;i++)for(j=0;j<2;j++)for(k=0;k<2;k++)(c.a[i][j]+=a[i][k]*b.a[k][j])%=1000000007;

    return c;

  }

}A,B,C;

int T,a[65],len,i,j,t;ll n,x,tmp,f[65][2][2];

int main(){

  scanf("%d",&T);

  while(T--){

    scanf("%lld",&x);

    for(len=0,tmp=x;tmp;a[++len]=tmp&1LL,tmp>>=1LL);

    for(i=1,j=len;i<=len&&i<j;i++,j--)t=a[i],a[i]=a[j],a[j]=t;

    for(i=1;i<=len;i++)f[i][0][0]=f[i][1][0]=f[i][0][1]=f[i][1][1]=0;

    for(i=0;i<=1;i++)f[1][i][i==a[1]]=1;

    for(i=1;i<len;i++){

      if(f[i][0][0])for(j=0;j<=1;j++)f[i+1][j][0]+=f[i][0][0];

      if(f[i][1][0])for(j=0;j<=0;j++)f[i+1][j][0]+=f[i][1][0];

      if(f[i][0][1])for(j=0;j<=a[i+1];j++)f[i+1][j][j==a[i+1]]+=f[i][0][1];

      if(f[i][1][1])for(j=0;j<=0;j++)f[i+1][j][j==a[i+1]]+=f[i][1][1];

    }

    printf("%lld\n",f[len][0][0]+f[len][1][0]+f[len][0][1]+f[len][1][1]-1);

    for(A=B=C=mat(),A.a[0][1]=A.a[1][0]=A.a[1][1]=B.a[0][0]=C.a[0][0]=C.a[1][1]=1,B.a[1][0]=2;x;x>>=1LL,A=A*A)if(x&1LL)C=C*A;

    C=C*B;

    printf("%lld\n",C.a[0][0]);

  }

  return 0;

}