零分求调

#include<bits/stdc++.h>
using namespace std;

const int N=1010;
int n,K;

struct T
{
    int a[N][N];
};

T mul(T a,T b)
{
    T c;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            for(int k=1;k<=n;k++)
            {
                c.a[i][j]=a.a[i][k]*b.a[k][j];
            }
        }
    }
    return c;
}

T power(T a,int b)
{
    T ans;
    for(int i=1;i<=n;i++)ans.a[i][i]=1;
    for(;b;b>>=1)
    {
        if(b&1)ans=mul(ans,a);
        a=mul(a,a);
    }
    return ans;
}

int main()
{
    T t;
    scanf("%d%d",&n,&K);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&t.a[i][j]);
        }
    }
    T res=power(t,K);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cout<<res.a[i][j]<<" ";
        }
        cout<<"\n";
    }
    return 0;
}