2020-12-05

矩阵快速幂

#include<iostream>
#include<cstring>
#define ll long long
#define maxn 105
#define mod 1000000007
using namespace std;
int n;ll k;
typedef struct{
    ll a[maxn][maxn];
    void build()
    {
        memset(a,0,sizeof(a));
    }
}matrix;

void init(matrix &x)
{
    x.build();
    for(ll i=0;i<n;i++)
    {
        x.a[i][i]=1;
    }
}

matrix input(matrix x)
{
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        cin>>x.a[i][j];
    }
    return x;
}

matrix operator *(const matrix x,const matrix y)
{
    matrix res;
    res.build();
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            for(int k=0;k<n;k++)
            res.a[i][j]=(res.a[i][j]+x.a[i][k]*y.a[k][j]%mod)%mod;
        }
    }
    return res;
}

matrix pow(matrix x)
{
    matrix s;
    init(s);
    while(k>0)
    {
        if(k&1)
        {
            s=s*x;
        }
        x=x*x;
        k>>=1;
    }
    return s;
}

int main()
{
    matrix x;
    cin>>n>>k;
    x=input(x);
    x=pow(x);
    for(int i=0;i<n;i++)
    {for(int j=0;j<n-1;j++)
    cout<<x.a[i][j]<<' ';
    cout<<x.a[i][n-1]<<endl;}
    return 0;

}