Week14—染砖

问题描述

衣食无忧的 Q老师 有一天突发奇想，想要去感受一下劳动人民的艰苦生活。

具体工作是这样的，有 N 块砖排成一排染色，每一块砖需要涂上红、蓝、绿、黄这 4 种颜色中的其中 1 种。且当这 N 块砖中红色和绿色的块数均为偶数时，染色效果最佳。

为了使工作效率更高，Q老师 想要知道一共有多少种方案可以使染色效果最佳，你能帮帮他吗？
【输入】
第一行为 T，代表数据组数。(1 ≤ T ≤ 100)

接下来 T 行每行包括一个数字 N，代表有 N 块砖。(1 ≤ N ≤ 1e9)
【输出】
输出满足条件的方案数，答案模 10007。
【样例输入】

2
1
2

【样例输出】

2
6

问题分析

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

转移方程如下：
• A[i] = 2 * A[i-1] + C[i-1]（乘2表示有除红绿以外黄蓝两种情况存在）
• 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 <string.h>
#include <stdio.h>
using namespace std;

struct Matrix{
	int x[3][3];
	Matrix operator*(Matrix&b)
	{
		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]*b.x[k][j]%10007;
					ret.x[i][j]%=10007;
				}
			}
		}
		return ret;
	}
	Matrix(){memset(x,0,sizeof(x));}
	Matrix(Matrix&t){memcpy(x,t.x,sizeof(x));}
};
Matrix quick_pow(Matrix a,int x)
{
	Matrix ret;
	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;
	cin>>T; 
	while(T--)
	{
		int k;
		cin>>k;
		if(k==1)
		{
			cout<<2<<endl;
			continue;
		}
		Matrix temp;
		temp.x[0][1]=temp.x[1][0]=0;
		temp.x[0][2]=temp.x[1][2]=1;
		temp.x[1][1]=temp.x[0][0]=temp.x[2][0]=temp.x[2][1]=temp.x[2][2]=2;
		temp=quick_pow(temp,k-1);
		int ans=0;
		ans=temp.x[0][0]*2%10007+temp.x[0][2]*2%10007;
		ans%=10007;
		cout<<ans<<endl;
	}
}