蓝桥杯 basic17 矩阵相乘

#include <iostream>
using namespace std;
int main()
{
    int n,m;
    cin>>n>>m;
    long long a[31][31]={0};
    long long b[31][31]={0};
    long long c[31][31]={0};
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            cin>>a[i][j];
            b[i][j]=a[i][j];
        }    
    int t=1    ;
    long long sum=0;
    if(m==0)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(i==j)
                    cout<<1<<" ";
                else
                    cout<<0<<" ";
            }
            cout<<endl;
        }
    }
    else
    {
        while (t<m)
        {
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=n;j++)
                {
                    sum=0;
                    for(int k=1;k<=n;k++)
                    {
                        sum+=a[i][k]*b[k][j];
                    
                    }
                    c[i][j]=sum;
                }
            }
                for(int i=1;i<=n;i++)
                    for(int j=1;j<=n;j++)
                    {
                        b[i][j]=c[i][j];
                    } 
            t++;    
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
                cout<<b[i][j]<<" ";
            cout<<endl;    
        }
    }

    return 0;
}
/*
    忘了线代的矩阵相乘，也忘了线代的矩阵零次方。 
*/