POJ - 3734 Blocks（矩阵快速幂）-CSDN博客

本文链接：https://blog.csdn.net/GYH0730/article/details/90369271

题意：个球，有红黄蓝绿4种颜色，要求红球和绿球有偶数个，有多少种排列方式？

思路：

$a_{i}$ ：红球与绿球都为偶数的个数

$b_{i}$ ：红球与绿球恰有一个为偶数的个数

$c_{i}$ ：红球与绿球都为奇数的个数

$a_{i+1}=2a_{i}+b_{i}$

$b_{i+1}=2a_{i}+2b_{i}+2c_{i}$

$c_{i+1}=b_{i}+2c_{i}$

#include <stdio.h>
#include <vector>
using namespace std;
const int mod = 10007;
typedef vector<int> vec;
typedef vector<vec> mat;

mat mul(mat &A,mat &B)
{
    int i,j,k;
    mat res( A.size(),vec(B[0].size()) );
    for(i = 0; i < A.size(); i++) {
        for(j = 0; j < B[0].size(); j++) {
            res[i][j] = 0;
            for(k = 0; k < A[0].size(); k++) {
                res[i][j] = ( res[i][j] + (A[i][k] * B[k][j]) % mod ) % mod;
            }
        }
    }
    return res;
}
mat q_pow(mat &A,int b)
{
    int i,j;
    mat res(A.size(),vec(A[0].size()));
    for(i = 0; i < A.size(); i++)
        res[i][i] = 1;
    while(b) {
        if(b & 1) res = mul(res,A);
        A = mul(A,A);
        b >>= 1;
    }
    return res;
}
int main(void)
{
    int T,n;
    mat base(3,vec(3));

    scanf("%d",&T);
    while(T--) {
        base[0][0] = 2,base[0][1] = 1,base[0][2] = 0;
        base[1][0] = 2,base[1][1] = 2,base[1][2] = 2;
        base[2][0] = 0,base[2][1] = 1,base[2][2] = 2;
        scanf("%d",&n);
        printf("%d\n",q_pow(base,n)[0][0]);
    }
    return 0;
}