【BZOJ 3329】[数位dp]Xorequ

题目描述

题目分析

首先我们发现$x \otimes 3x=2x \Rightarrow x\otimes 2x=3x$这便意味着$x\otimes (x<<1)$等于$x+(x<<1)$，我们可以发现，亦或是没有进位的加，而这里又相等，也就是说$x$与$x<<1$在同一位置上没有1，$x$换为二进制后没有相邻的两个1。
SubTask1就构造$dp_{i,0/1}$表示当有$i$位最高位为0/1的数的个数（允许前导0），有$dp_{i,0}=dp_{i-1,0}+dp_{i-1,1},dp_{i,1}=dp_{i,0}$。然后套路一下就行了。
SubTask2数据规模那么大，我们可以发现$2^{n}$二进制为$1000...000$，答案就是$dp_{n,1}=dp_{n-1,0}=dp_{n-2,0}+dp_{n-1,1}=dp_{n-2,0}+dp_{n-3,0}$，发现它是一个斐不拉几数列，呵呵，矩阵快速幂一下就行了。

代码

/**************************************************************
    Problem: 3329
    User: szpszp
    Language: C++
    Result: Accepted
    Time:32 ms
    Memory:1292 kb
****************************************************************/

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

#define MAXN
#define MAXM
#define INF 0x3f3f3f3f
typedef long long int LL;
const LL MOD = 1000000007ll;

template<class T>
void Read(T &x){
    x=0;char c=getchar();bool flag=0;
    while(c<'0'||'9'<c){if(c=='-')flag=1;c=getchar();}
    while('0'<=c&&c<='9'){x=x*10+c-'0';c=getchar();}
    if(flag)x=-x;
}

LL dp[70][2];
int num[70];
int len;

void getnum(LL x){
    len=0;
    while(x)num[++len]=x&1,x>>=1;
}

LL Solve1(LL x){
    getnum(x);

    LL ans=0;
    for(int i=1;i<len;++i)
        ans+=dp[i][1];

    for(int i=len-1;i>=1;--i){
        if(num[i]==1){
            ans+=dp[i][0];
            if(num[i+1]==1)break;
        }
    }

    return ans;
}

struct Matrix{
    LL data[2][2];
    int n,m;

    Matrix(){memset(data,0,sizeof(data));m=n=0;}
    Matrix(int a,int b){memset(data,0,sizeof(data));m=a,n=b;}
    void tobeunit(){
        for(int i=0;i<n;++i)data[i][i]=1;
    }
    Matrix operator * (const Matrix &x)const{
        Matrix rn=Matrix(m,x.n);
        if(n!=x.m)return rn;

        for(int i=0;i<m;++i)
            for(int j=0;j<x.n;++j)
                for(int k=0;k<n;++k)
                    rn.data[i][j]=(rn.data[i][j]+data[i][k]*x.data[k][j])%MOD;

        return rn;
    }

    Matrix operator *= (const Matrix &x){return (*this)=(*this)*x;}
};

Matrix ksm(Matrix a,LL p){
    Matrix rn=Matrix(2,2);
    rn.tobeunit();
    while(p){
        if(p&1)rn*=a;
        a*=a;
        p>>=1;
    }
    return rn;
}

LL Solve2(LL x){
    Matrix a,b;
    a=Matrix(1,2);
    a.data[0][0]=1,a.data[0][1]=1;
    b=Matrix(2,2);
    b.data[0][0]=0,b.data[0][1]=1;
    b.data[1][0]=1,b.data[1][1]=1;

    Matrix ans=a*ksm(b,x);
    return ans.data[0][1];
}

void init(){
    dp[1][0]=dp[1][1]=1;
    for(int i=2;i<=64;++i){
        dp[i][0]=dp[i-1][1]+dp[i-1][0];
        dp[i][1]=dp[i-1][0];
    }
}

int main(){
    init();

    int T;
    Read(T);

    LL x;
    while(T--){
        Read(x);
        printf("%lld\n%lld\n",Solve1(x+1),Solve2(x));
    }
}