题意:有大小为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;
}