【HDU6030】Happy Necklace(找规律+矩阵快速幂)

记录一个菜逼的成长。。

题目链接

题目大意:
给你n长度的项链，可以涂红可以涂蓝
满足连续的素数长度的子串里，红的数量不少于蓝。

看数据范围很容易想到找规律。
2个蓝色之间最少相隔2个红色
会发现
$f[0] = 1$
$f[1] = 2$
—————$f[0],f[1]$为补充———————
$f[2] = 3$
$f[3] = 4$
$f[4] = 6$
$f[5] = 9$
$f[6] = 13$
$f[7] = 19$
….
会发现
$f[n] = f[n-1] + f[n-3]$
存在递推式。考虑矩阵。
构造一个3*3的矩阵A
1 0 1
1 0 0
0 1 0
$f[n]\space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space f[2]$
$f[n-1] = A^{n-2} * f[1]$
$f[n-2]\space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space \space f[0]$

设$A = A^{n-2}$
$ans = f[n] = A[0][0] * f[2] + A[0][1] * f[1] + A[0][2] * f[0]$

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL MOD = 1e9 + 7;
typedef vector<LL> vec;
typedef vector<vec> mat;
LL n;
mat mul(mat &A,mat &B)
{
    mat C(A.size(),vec(B[0].size()));
    for( int i = 0; i < A.size(); i++ ){
        for( int k = 0; k < B.size();k++ ){
            for( int j = 0; j < B[0].size();j++ ){
                C[i][j] = (C[i][j] + A[i][k] * B[k][j]);
                C[i][j] %= MOD;
            }
        }
    }
    return C;
}
mat pow(mat A,LL n)
{
    mat B(A.size(),vec(A.size()));
    for( int i = 0; i < A.size(); i++ ){
        B[i][i] = 1;
    }
    while(n > 0){
        if(n & 1)B = mul(B,A);
        A = mul(A,A);
        n >>= 1;
    }
    return B;
}
void solve()
{
    mat A(3,vec(3));
    A[0][0] = A[0][2] = A[1][0] = A[2][1] = 1;
    A[0][1] = A[1][1] = A[1][2] = A[2][0] = A[2][2] = 0;
    A = pow(A,n-2);
    LL ans = (A[0][0] * 3 % MOD + A[0][1] * 2  % MOD + A[0][2] * 1) % MOD;
    printf("%lld\n",ans % MOD);
}
int main()
{
    int T;scanf("%d",&T);
    while(T--){
        scanf("%lld",&n);
        solve();
    }
    return 0;
}