题目大意:
给定方程x^3x=2x
求出在n以内的正整数解的个数,答案不模1e9+7
求出在2^n以内的正整数解的个数,答案模1e9+7
题目分析:
根据异或运算的性质:
a^b=c <=> a=b^c
所以原方程可改写为x^2x=3x
继续转换:x^2x=x+2x
根据a+b=(a^b)+((a&b)<<1)得
x^2x=(x^2x)+((x&2x)<<1)
即x&2x=0
即x&(x<<1)=0
则可以推出 x任意相邻二进制位不都为1 <=> x为该方程的解。
第一问可以用二进制数位dp做,f[i][j][p],i代表填到第i位,j代表填0还是填1,p代表是否是压着上限填。
第二问可以可以推出递推公式f[i]=f[i-1]+f[i-2](可以理解为在i-1位二进制数前填0,在i-2位二进制数前填10)菲波那切数列,矩阵乘法入门题。
感慨:
这道写起来不难,但是我第一遍却写挂了。
写数位dp和矩阵乘法这种糟心的东西,即使很水,果然也还是要心平气和,头脑清醒的时候去写。
否则会越写越糟心啊=。=
代码如下:
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
int T;
ll n;
struct squ{
ll a[2][2];
squ(ll a00=0,ll a01=0,ll a10=0,ll a11=0)
{
a[0][0]=a00; a[0][1]=a01;
a[1][0]=a10; a[1][1]=a11;
}
squ operator * (const squ &c)const
{
static squ z;
for(int i=0;i<2;i++)
for(int j=0;j<2;j++)
{
z.a[i][j]=0;
for(int k=0;k<2;k++)
z.a[i][j]=(z.a[i][j]+a[i][k]*c.a[k][j])%mod;
}
return z;
}
}I=squ(1,0,0,1),pre=squ(1,1,0,0),trans=squ(0,1,1,1);
squ Fast_Power(squ a,long long k)
{
squ ans=pre;
while(k)
{
if(k&1) ans=ans*a;
a=a*a;
k>>=1;
}
return ans;
}
ll f[64][2][2];
ll dp(ll n)
{
memset(f,0,sizeof(f));
f[63][0][1]=1;
for(int i=62;i>=0;i--)
for(int j=0;j<2;j++)
for(int p=0;p<2;p++)
{
int pp=0;
if(p==1 && j>((n>>i)&1)) continue;
if(p==1 && j==((n>>i)&1)) pp=1;
for(int k=0;k<2;k++)
if(!(k&j)) f[i][j][pp]+=f[i+1][k][p];
}
return f[0][0][1]+f[0][0][0]+f[0][1][0]+f[0][1][1]-1;
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%lld",&n);
ll ans1=dp(n);
squ ans2=Fast_Power(trans,n);
printf("%lld\n%lld\n",ans1,ans2.a[0][1]);
}
return 0;
}