【题意】给定方程x^3x=2x,求<=x和<=2^x的满足方程的正整数个数。
【算法】数位DP,矩阵快速幂
【题解】异或相当于不进位加法。
移项得,x^2x=3x,又因为x+2x=3x,所以x+2x不能产生进位。
又2x=x<<1,所以x+(x<<1)不进位当且仅当x中不存在相邻的1。
问题转化为求<=x的二进制不存在相邻1的正整数个数,state记录前一位为0或1,进行简单的数位DP。(递推的话就表示最高位,然后从下一位对应转移)
第二个问题,设2^x的答案为f[x],最高位为x。若x位为0,则ans=f[x-1]。若x位为1,则x-1位为0,ans=f[x-2]。
所以,f[x]=f[x-1]+f[x-2],用矩阵快速幂求解斐波那契数列。
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const int maxn=100,MOD=1e9+7; ll f[maxn][maxn],n,ans[5][5],A[5][5],c[5][5]; int a[maxn]; ll dfs(int pos,int state,int limit){ if(pos==-1)return 1; if(!limit&&~f[pos][state])return f[pos][state]; ll sum=dfs(pos-1,0,limit&&a[pos]==0); if((!limit||a[pos]==1)&&state==0)sum+=dfs(pos-1,1,limit); if(!limit)f[pos][state]=sum; return sum; } void mul(ll A[5][5],ll B[5][5]){ for(int i=0;i<2;i++) for(int j=0;j<2;j++){ c[i][j]=0; for(int k=0;k<2;k++) c[i][j]=(c[i][j]+B[i][k]*A[k][j])%MOD; } for(int i=0;i<2;i++) for(int j=0;j<2;j++) A[i][j]=c[i][j]; } ll ask(ll x){ ans[0][0]=ans[1][0]=1;ans[0][1]=ans[1][1]=0; A[0][0]=A[0][1]=A[1][0]=1;A[1][1]=0; while(x){ if(x&1)mul(ans,A); mul(A,A); x>>=1; } return ans[0][0]; } int main(){ int T; scanf("%d",&T); memset(f,-1,sizeof(f)); while(T--){ scanf("%lld",&n); ll num=n;// int len=0; while(num){ a[len++]=num%2; num/=2; } printf("%lld\n%lld\n",dfs(len-1,0,1)-1,ask(n)); } return 0; }
注意long long。