Codeforces 551D GukiZ and Binary Operations

problem

题意

给定n,l,k,m，假设一个长度为n的数组a，满足a[i] < $2^n$，并且有(a[1] and a[2])or(a[2] and a[3]) or ... (a[n-1] and a[n]) = k。问这样子的数组的个数有多少，答案模m。

---

思路

首先这道题我们要通过二进制位来计算。我们可以知道，对于每个位，可以独立计算。然后我们就可以计算每个位是0的种数。设dpi是长度为i的序列为0的种数。可以推出，dp[i] ＝ dp[i-1] + dp[i-2]所以我们用矩阵快速幂去求出斐波那契数列的值，然后用一下乘法原理即可。
写题解的原因是求斐波那契数列可以不用写矩阵快速幂。我们可以设f[n] = x * f[n -k ] + y * f[n - k -1]，对于k ＝ n/2时解出xy的值。这样我们就可以递归的去求解f[n]，时间复杂度仍为$O(logn)$。

AC代码

/*************************************************************************
    > File Name: pd.cpp
    > Author: znl1087
    > Mail: loveCJNforever@gmail.com 
    > Created Time: 六  6/20 09:11:46 2015
 ************************************************************************/

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <string>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#define LL long long
using namespace std;
LL n,k,l,m;
LL pow2(LL x){
    if( x == 1) return 2%m;
    LL temp = pow2(x/2);
    temp = (temp * temp)%m;
    if( x & 1)temp = (temp * 2)%m;
    return temp;
}
map<LL,LL> F;

LL fib(LL n){
    if(F.count(n))return F[n];
    LL k = n/2;
    if(n % 2 == 0) return F[n] = ( fib(k) * fib(k) % m + fib(k-1) * fib(k-1) % m)%m;
    else return F[n] = (fib(k) * fib(k+1) % m + fib(k-1) * fib(k) % m)%m;
}
int main(){
    cin>>n>>k>>l>>m;
    if( m == 1 || k >= (1ULL << min(l,63ll)) ){
        cout<<0<<endl;
        return 0;
    }
    F[0] = 1,F[1] = 1;
    LL res = 1;
    LL tot = pow2(n),zero = fib(n+1);
    LL temp = (tot - zero + m)%m;
    //cout<<tot<<endl;
    for(int i=0;i<l;i++){
        if(k & 1)res = res * temp % m;
        else res = res * zero %m;
        k>>=1;
    }
    cout<<res<<endl;
    return 0;
}