POJ 3734 Blocks 线性递推 矩阵快速幂

题意：有大小为1*N的格子，要给每个格子涂上红、蓝、绿、黄四种颜色。要求红色和绿色的格子的数目都是偶数。求最终的方案数。

思路：因为要求方案数，我们考虑是否是DP。因为N<=10^9，即使是线性的DP也会超时。下面我们就要想线性递推，复杂度刚好够。

           首先想到的是，黄色和蓝色是不影响红色和绿色填涂的。经过尝试，我们发现用ai:红色和绿色的格子都为偶数，bi：红色和绿色的格子一个为奇数一个为偶数，ci：红色和绿色的格子都是奇数个。

           下面考虑转移情况：

           1.对于ai. 可以从bi得到，即将将奇数的颜色在涂一个相同的颜色。 也可以从ai得到，即涂黄色和绿色。所以ai = bi + 2ai;

           2.对于bi，可以从ci得到，在红色和绿色中随便涂一个颜色；可以从bi得到，即涂黄色或蓝色；也可以从ai得到，同样涂红色或绿色。所以bi = 2ai + 2bi + 2ci；

           3.对于ci，可以从ci得到，涂黄色或蓝色；可以从bi得到，涂数目为偶数的格子。所以ci = 2ci + bi;

           这样，我们就能得到递推关系，将其在转化为矩阵即可。

代码如下：

#include <cstdio>
#include <vector>

using namespace std;

typedef vector<int> vec;
typedef vector<vec> mat;
typedef long long ll;

const int MOD = 10007;

mat mul(mat & A, mat &B)
{
	mat C(A.size(), vec(B[0].size()));
	for (int i = 0; i < A.size(); ++i)
		for (int k = 0; k < B.size(); k++)
			for (int j = 0; j < B[0].size(); ++j)
				C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % MOD;
	return C;
}
mat pow(mat A, ll n)
{
	mat B(A.size(), vec(A.size()));
	for (int i = 0; i < A.size(); ++i)
		B[i][i] = 1;
	while (n){
		if (n & 1) B = mul(B, A);
		A = mul(A, A);
		n >>= 1;
	}
	return B;
}

int N,T;

int main(void)
{
    //freopen("input.txt","r",stdin);
    scanf("%d",&T);
    while(T--){
        scanf("%d",&N);
        mat A(3,vec(3));
        A[0][0] = 2,A[0][1] = 1,A[0][2] = 0;
        A[1][0] = A[1][1] = A[1][2] = 2;
        A[2][0] = 0,A[2][1] = 1,A[2][2] = 2;
        A = pow(A,N);
        printf("%d\n",A[0][0]);
    }
    return 0;
}