poj 3233(矩阵)

/*
    poj 3233
    题目大意是
    给你矩阵A,求矩阵S=A+A^2+……+A^K
    
    一开始准备用矩阵求
    S(k)=S(k-1)+A^k
    然后就构造了一个由矩阵组成的矩阵
    C={E E}
      {0 A}
    后面感觉太麻烦了
    然后发现
    C^2={E E+A}
        {0 A^2}
    C^3={E E+A+A^2}
        {0   A^3  }
    也就是说
    C^(k+1)={E E+A+A^2+……+A^k}
            {0      A^3      }
    然后发现右上角那一块跟要求的S很像
    所以我们算出C^(k+1)之后
    把右上角的一块减去单位矩阵E就行
    
*/
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <math.h>
#include <string.h>
#define ll long long
using namespace std;
struct matrix
{
    ll m[65][65];//因为我们之后要构建一个由矩阵建立的矩阵，所以一开始设数组时一定要是上限的2倍往上。
};
ll nn,k,mod;
matrix matrixmul(matrix a,matrix b)
{
    matrix c;
    for(int i=1; i<=nn; i++)
        for(int j=1; j<=nn; j++)
        {
            c.m[i][j]=0;
            for(int k=1; k<=nn; k++)
                c.m[i][j]+=(a.m[i][k]*b.m[k][j])%mod;
            c.m[i][j]%=mod;
        }
    return c;
}
matrix quickpow(matrix m,ll n)
{
    matrix b;
    memset(b.m,0,sizeof(b.m));
    for(int i = 1; i <=nn; i++)
        b.m[i][i] = 1;
    while(n>=1)
    {
        if(n&1)
            b=matrixmul(b,m);
        n=n>>1;
        m=matrixmul(m,m);
    }
    return b;
}


int main()
{
    cin>>nn>>k>>mod;
    matrix res;
    memset(res.m,0,sizeof(res.m));//使大矩阵左下角成为0矩阵
    for(int i=1; i<=nn; i++)//将大矩阵上半部分写成两个单位阵
        res.m[i][i]=res.m[i][i+nn]=1;
    for(int i=1; i<=nn; i++)//将给的矩阵A直接放入大矩阵右下角
        for(int j=1; j<=nn; j++)
            cin>>res.m[i+nn][j+nn];
    nn*=2;//因为是要对由矩阵构建的矩阵做乘法，所以乘的项应该是nn的两倍
    matrix ans=quickpow(res,k+1);
    nn/=2;//后面因为要从大矩阵中，剥离出一个nn*nn的小矩阵，所以要将开始乘2的结果除以2
    for(int i=1;i<=nn;i++)//将右上角减去单位矩阵
        ans.m[i][i+nn]--;
    for(int i=1;i<=nn;i++)//输出
    {
        for(int j=1;j<nn;j++)
            cout<<(ans.m[i][j+nn]+mod)%mod<<' ';//因为前面有些项减过1有可能成为负数,所以还要进行一次取模
        cout<<(ans.m[i][2*nn]+mod)%mod<<endl;
    }


    return 0;
}