Q老师染砖【动态规化dp】【矩阵快速幂优化】

问题描述

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

Input

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

Output

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

Sample Input

2
1
2

Sample Output

2
6

问题分析

连续格子染色，很明显有子结构的性质，考虑 DP

构造DP状态
A[i] 表示 i 个格子，红绿均为偶数的染色方案数
B[i] 表示 i 个格子，红绿均为奇数的染色方案数
C[i] 表示 i 个格子，红绿有一个为偶数的染色方案数

DP 转移方程
• 根据 DP 状态，仔细思考转移情况即可得到
• 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]

但是 n 很大，因此考虑矩阵快速幂优化 DP

复杂度：O(3^3 * log(n)) = O(log (n))

#include<stdio.h>
#include<string.h>
using namespace std;
struct Matrix{
	int x[3][3];
	Matrix operator*(const Matrix&t)const{
		Matrix ret;
		for(int i=0;i<3;i++)
			for(int j=0;j<3;j++){
				ret.x[i][j]=0;
				for(int k=0;k<3;k++){
					ret.x[i][j]+=x[i][k]*t.x[k][j]%10007;
					ret.x[i][j]%=10007;
				}
			}
		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;
	memset(ret.x,0,sizeof ret.x);
	for(int i=0;i<3;i++)
		ret.x[i][i]=1;
	while(x){
		if(x&1)ret=ret*a;
		a=a*a;
		x>>=1;
	}
	return ret;
}
int main(){
	int T,N;
	scanf("%d",&T);
	while(T--){
		scanf("%d",&N);
		Matrix a;
		a.x[0][0]=a.x[1][1]=a.x[2][0]=a.x[2][1]=a.x[2][2]=2;
		a.x[0][2]=a.x[1][2]=1;
		Matrix f=quick_pow(a,N-1);
		int ans=f.x[0][0]*2+f.x[0][2]*2;
		ans%=10007;
		printf("%d\n",ans);
	}
	
	return 0;
}