poj3233矩陣的冪的和

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<vector>

using namespace std;
int n,k,m;
typedef vector<int> vec;
typedef vector<vec> mat;
mat mul(mat &a,mat &b)
{
    mat c(a.size(),vec(b[0].size()));
    for(int i=0;i<a.size();i++)
    {
        for(int k=0;k<b.size();k++)
        {
            for(int j=0;j<b[0].size();j++)
            {
                c[i][j]=(c[i][j]+a[i][k]*b[k][j])%m;
            }
        }
    }
    return c;
}
mat pow(mat a,int n)
{
    mat b(a.size(),vec(a.size()));
    for(int i=0;i<a.size();i++)
        b[i][i]=1;
    while(n)
    {
        if(n&1)
            b=mul(b,a);
        a=mul(a,a);
        n>>=1;
    }
    return b;
}
int main()
{
    scanf("%d%d%d",&n,&k,&m);
    mat b(n*2,vec(n*2));
    int t;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            scanf("%d",&t);
            b[i][j]=t;
        }
        b[n+i][i]=b[n+i][i+n]=1;
    }
    b=pow(b,k+1);
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            int a=b[i+n][j]%m;
            if(i==j)
                a=(a+m-1)%m;
            printf("%d%c",a,j+1==n?'\n':' ');
        }
    }
    return 0;
}