ACM训练赛赛后补题：Happy Necklace（思维+递推+矩阵快速幂）

题目描述：


分析

这道题很容易就可以定性为动态规划，需要能够推出递推公式；然后观察发现n太大了，最多只能接收O(logn)的复杂度，这样的复杂度，实现的方式就是矩阵快速幂。

首先题目所说的是这一串项链里面的任何一个长度为素数的子串，都必须满足红色珠子的个数大于等于蓝色珠子的个数，那么我们可以粗略的将dp数组的状态定义为两个，f[i][j]，其中i表示当前链子的长度为i，j表示这一串链子的最后一个元素（即下标i）的颜色（0表示蓝色，1表示红色）
容易得知：

（1）f[i][1]=f[i-1][1]+f[i-1][0]，不论前一个珠子是红色还是蓝色，只要放上红色珠子都是可以的；
（2）f[i][0]=f[i-2][1]。可以这样想，想要在这里放上蓝色珠子，那么显然前两个元素都得是红色的珠子，因为不管是长度为2还是长度为3都必须要是红珠子数量>=蓝珠子数量，可以看做是长度为i-2，且最后一个元素为红色珠子的基础上，再加上“红蓝”这样的情况。
这里其实是有一些思维的，具体体现在，当我们在加入一个蓝色珠子的时候，前面的长度为素数的子串情况明明有很多，但是我们只考虑了长度为2的情况，这是因为，除了2以外的所有的素数，都是奇数，既然前面满足了限定条件，那么可以推断红色珠子的数量一定大于蓝色珠子的数量，因此我们只需要考虑i-2这个情况，其他的任何情况加上蓝色珠子都会满足“素数长度的子串中红色数量>=蓝色数量”。

接下来记录ans[i]=f[i][0]+f[i][1]，易得
ans[i]=f[i-2][1]+f[i-1][1]+f[i-1][0]=(f[i-3][1]+f[i-3][0])+(f[i-1][1]+f[i-1][0])
=ans[i-3]+ans[i-1]

因此最后这题我们的总递推公式就记为：

f[n]=f[n-1]+f[n-3]

但是我们可以发现这题的范围太大了，也不可能开这么大数组，而且常规的递推下去是会超时的，因此考虑矩阵快速幂来实现加速，复杂度是O(logn)，轻松过掉这道题。

因此我们构造一下这个矩阵，通过递推公式易得：

可以用快速幂的降幂思维（logn），容易得知，当幂次为奇数的时候，直接乘以这个3*3矩阵即可进入下一级别，当为偶数的时候，需要将2个原3*3矩阵先进行相乘，因为乘法运算律同样适用于矩阵，只要矩阵的相乘合理即可，不多解释了，推一下就行。

接下来就是我们的初始化问题，初始的矩阵应该设置为(f[3],f[2],f[1])，f[1]本身不存在，但是我们可以根据递推公式对其进行构造，由于f[4]=f[3]+f[1]，推出f[1]=2，如此初始化即可。

代码如下：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int MOD=1e9+7;
struct Matrix{
    ll a[5][5];
    Matrix(){memset(a,0,sizeof(a));}
};
inline Matrix Mul(Matrix a,Matrix b){
    Matrix temp;
    for(int i=1;i<=3;++i)
        for(int j=1;j<=3;++j)
            for(int k=1;k<=3;++k)
                temp.a[i][j]=(temp.a[i][j]+(a.a[i][k]*b.a[k][j])%MOD)%MOD;
    return temp;
}
inline int fun(ll n){
    Matrix ans,res;
    ans.a[1][1]=4;ans.a[1][2]=3;ans.a[1][3]=2;
    res.a[1][1]=res.a[1][2]=res.a[2][3]=res.a[3][1]=1;
    while(n){//快速幂思想
        if(n&1)ans=Mul(ans,res);
        res=Mul(res,res);
        n>>=1;
    }
    return ans.a[1][1];
}
int main(){
    int t;cin>>t;
    while(t--){
        ll n;scanf("%lld",&n);
        if(n==2)printf("3\n");
        else if(n==3)printf("4\n");
        else printf("%d\n",fun(n-3));
    }
}