题意:
衣食无忧的 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
思路:
定义状态A[i],B[i],C[i]
分别为前i块砖
的情况下,红绿块数全为偶数、全为奇数、一奇一偶的方案数。
不难看出A[1]=2,B[1]=0,C[1]=2
。
转移过程:
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的范围很大,O(n)的复杂度是无法接受的,可以采用矩阵快速幂进行优化。
矩阵递推式如下所示:
基于矩阵乘法的结合律,累乘可得:
利用矩阵快速幂求得常数矩阵的n-1次方,从而便可得到答案。
代码有注释。
代码:
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 1e9+5;
const int N=4;
int n,mod= 10007;
struct Matrix
{
int m[N][N];
void ini()
{//初始化A
memset(m, 0, sizeof(m));
m[1][3]=m[2][3]=1;
m[1][1]=m[2][2]=m[3][1]=m[3][2]=m[3][3]=2;
}
void ini1()
{//初始化F(1)
memset(m, 0, sizeof(m));
m[1][1]=2;
m[2][1]=0;
m[3][1]=2;
}
Matrix() {}
Matrix(const Matrix &t)
{
memcpy(m, t.m, sizeof(m));
}
Matrix operator*(const Matrix &t) const
{//乘法
Matrix res;
for (int i = 1; i < N ; i++)
{
for (int j = 1; j < N; j++)
{
res.m[i][j] = 0;
for (int k = 1; k < N; k++)
{
res.m[i][j] += (m[i][k] * t.m[k][j])%mod;
}
res.m[i][j]%=mod; //注意取模
}
}
return res;
}
};
Matrix quick_pow(Matrix a, Matrix b, int x)
{ //快速幂
while (x)
{
if (x & 1)
{
b = a * b;
}
a = a * a;
x >>= 1;
}
return b;
}
Matrix A, f;
int main()
{
int T;scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
A.ini();
f.ini1();
f=quick_pow(A,f,n-1);
printf("%d\n",f.m[1][1]);
}
return 0;
}