动态规划（矩阵快速幂优化DP，week14作业）

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;
}