[蓝桥杯][基础练习VIP]矩阵乘法

//eg.注意如何处理m=0时候的情况，需要进行特判
#include<cstring>
#include<cstdio>
#include<iostream>
using namespace std;
const int maxn = 31;
typedef long long LL;
LL A[maxn][maxn],B[maxn][maxn],C[maxn][maxn],E[maxn][maxn];
int main(void)
{
    for(int i=1;i<=30;i++) E[i][i]=1;
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            cin>>A[i][j];
            B[i][j]=C[i][j]=A[i][j];
        }
    if(m==0)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                cout<<E[i][j]<<" ";
            }
            cout<<endl;
        }
    }
    else
    {
        for(int k=0;k<m-1;k++)
        {
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=n;j++)
                {
                    LL sum=0;
                    for(int t=1;t<=n;t++)
                    {
                        sum+=A[i][t]*B[t][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];
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                cout<<C[i][j]<<" ";
            }
            cout<<endl;
        }
    }
    
    return 0;
}