POJ 3734 Blocks 矩阵快速幂

题意：给N个方块染色，每个方块可以染红蓝绿黄四种颜色，红色&&绿色方块数必须为偶数，求方案数（%10007）。

首先有DP方程：
$a_{i+1}=2*a_{i}+1*b_i+0*c_i$(a为红绿均为偶数的方案数)
$b_{i+1}=2*a_{i}+2*b_i+2*c_i$(b为红绿一奇一偶的方案数)
$c_{i+1}=0*a_{i}+1*b_{i}+2*c_i$(c为红绿均为奇数的方案数)
所以有矩阵A：
{2,2,0}
{1,2,1}
{0,2,2}
答案即{1,0,0}*$A^n$
直接乘是O(N*T),会超时，so 用一个快速幂优化即可~~~

#include <cstdio>
#include <cstring>
#define mod 10007
using namespace std;
int n,T;
struct xx
{
    int a[3][3];
}a,ans,tmp;

xx mul(xx a,xx b)
{
    xx tmp;
    memset(tmp.a,0,sizeof(tmp.a));
    for (int i=0;i<3;++i)
      for (int j=0;j<3;++j)
        for (int k=0;k<3;++k)
          tmp.a[i][j]=(tmp.a[i][j]+a.a[i][k]*b.a[k][j])%mod;
    return tmp;
}

int qsm(int x)
{
    if (x==0) return 0;
    memset(ans.a,0,sizeof(ans.a));
    ans.a[0][0]=1;
    //a=(xx){{2,2,0},{1,2,1},{0,2,2}};
    a.a[0][0]=2;a.a[0][1]=2;a.a[0][2]=0;
    a.a[1][0]=1;a.a[1][1]=2;a.a[1][2]=1;
    a.a[2][0]=0;a.a[2][1]=2;a.a[2][2]=2;
    xx tmp=a;
    x--;
    while (x)
    {
        if (x&1) tmp=mul(tmp,a);
        a=mul(a,a);
        x>>=1;
    }
    ans=mul(ans,tmp);
    return ans.a[0][0];
}

int main()
{
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d",&n);
        printf("%d\n",qsm(n));
    }
}