POJ 3233 Matrix Power Series（矩阵乘法）

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<map>
#include<iomanip>
#define INF 99999999
using namespace std;
const int MAX=30+10;
__int64 array[MAX][MAX],sum[MAX][MAX],temp[MAX][MAX],ans[MAX][MAX];
int n,m,k;
void MatrixInit(__int64 a[MAX][MAX],bool flag)
{
    for(int i=0; i<n; ++i)
    {
        for(int j=0; j<n; ++j)
        {
            if(flag)a[i][j]=array[i][j];
            else a[i][j]=(i == j);
        }
    }
}

void MatrixAdd(__int64 a[MAX][MAX],__int64 b[MAX][MAX],int &mod)
{
    for(int i=0; i<n; ++i)
    {
        for(int j=0; j<n; ++j)
        {
            a[i][j]=(a[i][j]+b[i][j])%mod;
        }
    }
}

void MatrixMult(__int64 a[MAX][MAX],__int64 b[MAX][MAX],int &mod)
{
    __int64 c[MAX][MAX]= {0};
    for(int i=0; i<n; ++i)
    {
        for(int j=0; j<n; ++j)
        {
            for(int k=0; k<n; ++k)
            {
                c[i][j]+=a[i][k]*b[k][j];
            }
        }
    }
    for(int i=0; i<n; ++i)
    {
        for(int j=0; j<n; ++j)a[i][j]=c[i][j]%mod;
    }
}
void MatrixPow(int k,int &mod) //矩阵快速幂
{
    MatrixInit(sum,0);
    MatrixInit(temp,1);
    while(k)
    {
        if(k&1)MatrixMult(sum,temp,mod);
        MatrixMult(temp,temp,mod);
        k>>=1;
    }
}
void MatrixSum(int k,int &mod) //矩阵和
{
    if(k == 1)
    {
        MatrixInit(ans,1);
        return;
    }
    MatrixSum(k/2,mod);
    MatrixPow((k+1)/2,mod);
    if(k&1)
    {
        MatrixInit(temp,1);
        MatrixAdd(sum,temp,mod);
        MatrixMult(ans,sum,mod);
        MatrixAdd(ans,temp,mod);
    }
    else
    {
        MatrixInit(temp,0);
        MatrixAdd(temp,sum,mod);
        MatrixMult(ans,temp,mod);
    }
}
int main()
{
    scanf("%d%d%d",&n,&k,&m);
    for(int i=0; i<n; ++i)
    {
        for(int j=0; j<n; ++j)
        {
            scanf("%I64d",&array[i][j]);
            ans[i][j]=0;
        }
    }
MatrixSum(k,m);
    for(int i=0; i<n; ++i)
    {
        for(int j=0; j<n-1; ++j)
            printf("%d ",ans[i][j]);
        printf("%I64d\n",ans[i][n-1]);
    }
    return 0;
}