HDU 6470 【矩阵快速幂】

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6470

写这道题是为了让自己不要忘记矩阵快速幂如何推出矩阵式子的。

注意 代码是TLE的！！

#include<stdio.h>
#include<string.h>
#define mem(a, b) memset(a, b, sizeof(a))
typedef long long ll;
const int mod = 123456789;

ll n;

struct Matrix
{
    ll a[10][10];
}A, res, temp;

Matrix Multiply(Matrix a, Matrix b)
{
    Matrix ans;
    mem(ans.a, 0);
    for(int i = 1; i <= 6; i ++)
        for(int j = 1; j <= 6; j ++)
            for(int k = 1; k <= 6; k ++)
            {
                ans.a[i][j] += a.a[i][k] * b.a[k][j] % mod;
                ans.a[i][j] %= mod;
            }
    return ans;
}

int main()
{
    mem(A.a, 0);  //构造矩阵 
    A.a[1][2] = A.a[2][2] = A.a[2][3] = A.a[3][3] = A.a[3][6] = A.a[4][4] = A.a[4][6] = A.a[5][5] = A.a[5][6] = A.a[6][6] = 1;
    A.a[2][1] = A.a[4][5] = 2;
    A.a[3][4] = A.a[3][5] = 3;
    int T;
    scanf("%d", &T);
    while(T --)
    {
        temp = A;  //初始化temp矩阵 
        mem(res.a, 0); //初始化为单位矩阵 
        for(int i = 1; i <= 6; i ++)
            res.a[i][i] = 1;
        scanf("%lld", &n);
        n -= 2;
        while(n)
        {
            if(n % 2)
                res = Multiply(res, temp);
            temp = Multiply(temp, temp);
            n /= 2;
        }
        printf("%lld\n", (1 * res.a[2][1] % mod + 2 * res.a[2][2] % mod + 27 * res.a[2][3] % mod + 9 * res.a[2][4] % mod + 3 * res.a[2][5] % mod + res.a[2][6] % mod) % mod);
    }
    return 0;
}