D - Q老师染砖(选做)
题目
衣食无忧的 Q老师 有一天突发奇想,想要去感受一下劳动人民的艰苦生活。
具体工作是这样的,有 N 块砖排成一排染色,每一块砖需要涂上红、蓝、绿、黄这 4 种颜色中的其中 1 种。且当这 N 块砖中红色和绿色的块数均为偶数时,染色效果最佳。
为了使工作效率更高,Q老师 想要知道一共有多少种方案可以使染色效果最佳,你能帮帮他吗?
第一行为 T,代表数据组数。(1 ≤ T ≤ 100)
接下来 T 行每行包括一个数字 N,代表有 N 块砖。(1 ≤ N ≤ 1e9)
输出满足条件的方案数,答案模 10007
2
1
2
2
6
思路
矩阵快速幂:
设A为n*n方阵,k为一个常数,若求A^k时间超限时,可以借鉴快速幂的方法进行求解,复杂度由O(N0缩短为O(logN)。
快速幂具体为:
而将线性递推式(这里以斐波那契为例)转化为矩阵快速幂求解的方法如下图所述:
本题解析:
如下图所示,列出状态方程后,转化为矩阵快速幂,然后求解。
代码
#include<iostream>
#include<algorithm>
#include<memory.h>
using namespace std;
const int maxn=1e9+50;
const int N=3;
struct matrix
{
int x[N][N];
matrix operator*(const matrix& t)
{
matrix res;
for(int i=0;i<N;i++)
for(int j=0;j<N;j++)
{
res.x[i][j]=0;
for(int k=0;k<N;k++)
res.x[i][j]+=x[i][k]*t.x[k][j]%10007;
res.x[i][j]%=10007;
}
return res;
}
matrix()
{
memset(x,0,sizeof x);
}
matrix(const matrix& t)
{
memcpy(x,t.x,sizeof x);
}
};
matrix quick_pow(matrix a,int k)
{
matrix res;
for(int i=0;i<N;i++)res.x[i][i]=1;
while(k)
{
if(k&1)res=res*a;
a=a*a;
k>>=1;
}
return res;
}
void solve(int n)
{
/*a[i]=2*a[i-1]+c[i-1];
b[i]=2*b[i-1]+c[i-1];
c[i]=2*a[i-1]+2*b[i-1]+2*c[i-1];*/
matrix a;
a.x[0][0]=2,a.x[0][1]=0,a.x[0][2]=1;
a.x[1][0]=0,a.x[1][1]=2,a.x[1][2]=1;
a.x[2][0]=2,a.x[2][1]=2,a.x[2][2]=2;
a=quick_pow(a,n-1);
int ans=0;
ans=a.x[0][0]*2+a.x[0][1]*0+a.x[0][2]*2;
ans%=10007;
cout<<ans<<endl;
}
int main()
{
int t;
cin>>t;
while(t--)
{
int n;
cin>>n;
solve(n);
}
return 0;
}