POJ - 3734 Blocks（矩阵快速幂）

题目大意：给出n个排成一列的方块，用红、蓝、绿、黄四种颜色给它们染色，要求出染成红色的方块个数和染成绿色的方块个数同时为偶数的染色方案数为多少，并对10007取模。

题解：

从左边开始考虑，染第i个格子的时候有三种情况：

1. 之前i-1个方块中，红色的方块个数和绿色的方块个数均为偶数，这时想要满足条件，就要染蓝色或黄色；

2. 之前i-1个方块中，红色的方块个数和绿色的方块个数中有一个为奇数，这时想满足条件就要染奇数个对应的颜色；

3. 之前i-1个方块中，红色的方块个数和绿色的方块个数均不是偶数，这时无法满足条件。

a[i+1]=2*a[i]+b[i] 
b[i+1]=2*a[i]+2*b[i]+2*c[i]
c[i+1]=b[i]+2*c[i]

即

我们最后要求的是n个方块红色和绿色个数都为偶数的染色方案数，即An

用矩阵快速幂来求

#include<cstring>
#include<iostream>
#include<cstdio>
using namespace std;
#define ll long long
#define mod 10007
int k;
int dim=3;//矩阵维数
struct matirx//注意矩阵的大小
{
    ll ma[dim+10][dim+10];
};
matirx mutimatirx(matirx a,matirx b)//矩阵乘
{
    matirx t;
    memset(t.ma,0,sizeof(t.ma));
    for(int i=0; i<dim; i++)
    {
        for(int k=0; k<dim; k++)
        {
            for(int j=0; j<dim; j++)
            {
                t.ma[i][j]+=a.ma[i][k]*b.ma[k][j];
                t.ma[i][j]%=mod;
            }
        }
    }
    return t;
}
matirx powmatirx(matirx a,long long b)//矩阵快速幂
//返回矩阵a的b次方
{
    matirx t;
    memset(t.ma,0,sizeof(t.ma));
    for(int i=0; i<dim; i++)
        t.ma[i][i]=1;
    while(b)
    {
        if(b&1)
        {
            t=mutimatirx(t,a);
        }
        a=mutimatirx(a,a);
        b/=2;
    }
    return t;
}
int main()
{
    int T;
    cin>>T;
    ll n;
    while(T--)
    {
        cin>>n;
        matirx res;
        memset(res.ma,0,sizeof(res.ma));
        res.ma[0][0]=2;res.ma[0][1]=1;res.ma[0][2]=0;
		res.ma[1][0]=2;res.ma[1][1]=2;res.ma[1][2]=2;
		res.ma[2][0]=0;res.ma[2][1]=1;res.ma[2][2]=2;
        res=powmatirx(res,n);//求res矩阵的n次幂
        cout<<res.ma[0][0]<<endl;
    }
    return 0;
}