记录一个菜逼的成长。。
题目大意:
给你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] f[2]
f[n−1]=An−2∗f[1]
f[n−2] f[0]
设
A=An−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;
}