BZOJ_P3329 Xorequ(矩阵乘法+动态规划)

BZOJ传送门
Time Limit: 1 Sec Memory Limit: 256 MB
Submit: 428 Solved: 187
[Submit][Status][Discuss]
Description

Input
第一行一个正整数，表示数据组数据 ，接下来T行
每行一个正整数N

Output
2*T行
第2*i-1行表示第i个数据中问题一的解，
第2*i行表示第i个数据中问题二的解，

Sample Input
1
1

Sample Output
1
2

HINT
x=1与x=2都是原方程的根，注意第一个问题的解
不要mod 10^9+7
1<=N<=10^18
1<=T<=1000

Source
By Wcmg

题解:
题目x^3x=2x
移项得:x^2x=3x
即x^(x<<1)=x+(x<<1)
所以那必然啊:x&(x<<1)=0
然后递推，递推关系只与末项有关
试求f[i][0]=f[i-1][0]+f[i-1][1],f[i][1]=f[i-1][0]
可以变成一维f[i]=f[i-1]+f[i-2]，Fibonacci数列，2^n好算，矩阵乘就好
[0,n]的计算如果先转2进制，如果遇到两个连续的1辣么后面的都不用算了

#include<cstdio>
#include<vector>
using namespace std;
#define Mod 1000000007
inline long long in(long long x=0,char ch=getchar(),int v=1){
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();if(ch=='-') v=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x*v;}
typedef vector<long long> vec;
typedef vector<vec> mat;
mat mul(mat &A,mat &B){
    mat C(2,vec(2));
    for(int i=0;i<2;i++)
        for(int k=0;k<2;k++)
            for(int j=0;j<2;j++)
                C[i][j]=(C[i][j]+A[i][k]*B[k][j])%Mod;
    return C;
}
mat pow(mat A,long long b){
    mat res(2,vec(2));
    res[0][0]=res[1][1]=1,res[0][1]=res[1][0]=0;
    while(b){
        if(b&1) res=mul(res,A);
        A=mul(A,A);b>>=1;
    }
    return res;
}
long long f[65],w[65];int s;
long long A(long long x,long long res=0){
    s=0;while(x) w[++s]=(x&1),x>>=1;
    w[s+1]=0;
    for(int i=s;i;i--){
        if(w[i]) res+=f[i+1];
        if(w[i]&&w[i+1]) break;
    }
    return res-1;
}
mat a(2,vec(2)),b(2,vec(2));
int main(){
    int t;long long x;t=in();
    a[0][0]=a[1][0]=a[0][1]=1,a[1][1]=0;f[0]=0,f[1]=1;
    for(int i=2;i<65;i++) f[i]=f[i-1]+f[i-2];
    while(t--){
        x=in();b=pow(a,x+1);
        printf("%lld\n",A(x+1));
        printf("%lld\n",b[0][0]);
    }
    return 0;
}