UVA 11149 - Power of Matrix（矩阵乘法）

求矩阵幂次的和
就是让求：$A^1 + A^2 + A^3 + \cdots + A^n$
如果我们一个一个求A的幂次，再加起来，算一下复杂度，肯定会超时。
我们可以转化一下：
原式 = $A^1 + A^2 + \cdots + A^{n/2} + A^{n/2}*(A^1 + A^2 + \cdots + A^{n/2})$

= $(E+A^{n/2})*(A^1 + A^2 + \cdots + A^{n/2} )$

这样我们就可以用递归求出$A^1 + A^2 + A^3 + \cdots + A^n$ 的值了。
还有要注意的是当n为奇数时，n/2+n/2 = n-1，所以要在后面加上$A^n$ 。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
using namespace std;

const int mod = 10;

struct Matrix
{
    long long m[40][40];
    int n;
    Matrix(int x)
    {
        n = x;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                m[i][j] = 0;
    }
    Matrix(int _n,int a[40][40])
    {
        n = _n;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
            {
                m[i][j] = a[i][j];
            }
    }
};
Matrix operator *(Matrix a,Matrix b)
{
    int n = a.n;
    Matrix ans = Matrix(n);
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            for(int k=0;k<n;k++)
            {
                ans.m[i][j] += (a.m[i][k]%mod)*(b.m[k][j]%mod)%mod;
                ans.m[i][j] %= mod;
            }
    return ans;
}
Matrix operator ^(Matrix a,int k)
{
    int n = a.n;
    Matrix c(n);
    int i,j;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            c.m[i][j] = (i==j);
    for(;k;k>>=1)
    {
        if(k&1)
            c=c*a;
        a = a*a;
    }
    return c;
}

Matrix operator +(Matrix a,Matrix b)
{
    int n = a.n;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            a.m[i][j] = (a.m[i][j] + b.m[i][j])%mod;
    return a;
}
Matrix solve(Matrix a,int k)
{
    if(k == 1)
        return a;
    int n = a.n;
    Matrix E(n);
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            E.m[i][j] = (i==j);
    Matrix res(n);
    res = (E+(a^(k/2)))*solve(a,k/2);
    if(k & 1)
        res = res + (a^k);
    return res;
}
int main(void)
{
    int n,k,i,j;
    while(scanf("%d%d",&n,&k)==2)
    {
        if(n == 0)
            break;
        Matrix ans(n);
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
            {
                scanf("%d",&ans.m[i][j]);
                ans.m[i][j] %= mod;
            }
        ans = solve(ans,k);
        for(i=0;i<n;i++)
        {
            for(j=0;j<n-1;j++)
                printf("%d ",ans.m[i][j]);
            printf("%d\n",ans.m[i][j]);
        }
        printf("\n");
    }


    return 0;
}