SDU程序设计思维与实践Week14 Q老师染砖【矩阵快速幂优化DP】

Q老师染砖【POJ 3734 挑战程序设计竞赛P202】

衣食无忧的 Q老师 有一天突发奇想，想要去感受一下劳动人民的艰苦生活。
具体工作是这样的，有 N 块砖排成一排染色，每一块砖需要涂上红、蓝、绿、黄这 4 种颜色中的其中 1 种。且当这 N 块砖中红色和绿色的块数均为偶数时，染色效果最佳。
为了使工作效率更高，Q老师 想要知道一共有多少种方案可以使染色效果最佳，你能帮帮他吗？

输入

第一行为 T，代表数据组数。(1 ≤ T ≤ 100)
接下来 T 行每行包括一个数字 N，代表有 N 块砖。(1 ≤ N ≤ 1e9)

输出

输出满足条件的方案数，答案模 10007。

输入样例

2
1
2

输出样例

2
6

思路分析

• A[i] 表示 i 个格子，红绿均为偶数的染色方案数
• B[i] 表示 i 个格子，红绿均为奇数的染色方案数
• C[i] 表示 i 个格子，红绿一奇一偶的染色方案数
• 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]
• 转移矩阵

解题代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=3,m=10007;
int T,n;
struct Matrix{
	int x[N][N];
	Matrix operator*(const Matrix& t) const{
		Matrix ret;
		for(int i=0;i<N;i++){
			for(int j=0;j<N;j++){
				ret.x[i][j]=0;
				for(int k=0;k<N;k++){
					ret.x[i][j]+=x[i][k]*t.x[k][j]%m;
					ret.x[i][j]%=m;
				}
			}
		}
		return ret;
	} 
	Matrix(){
		memset(x,0,sizeof x);
	}
	Matrix(const Matrix&t){
		memcpy(x,t.x,sizeof x);
	}
};
Matrix quick_pow(Matrix a,int x){
	Matrix ret;
	for(int i=0;i<N;i++){
		ret.x[i][i]=1;
	}	
	while(x){
		if(x&1) ret=ret*a;		
		a=a*a;
		x>>=1;
	}
	return ret;
} 
int main(){
	cin>>T;
	while(T--){
		cin>>n;
		Matrix trans;
		trans.x[0][0]=2,trans.x[0][1]=0,trans.x[0][2]=1;
		trans.x[1][0]=0,trans.x[1][1]=2,trans.x[1][2]=1;
		trans.x[2][0]=2,trans.x[2][1]=2,trans.x[2][2]=2;
		trans=quick_pow(trans,n-1);
		int ans=0;
		ans+=trans.x[0][0]*2%m;ans%=m;
		ans+=trans.x[0][2]*2%m;ans%=m;
		cout<<ans<<endl;
	}
}