蓝桥杯基础练习 矩阵乘法

ans数组为最后需要输出的答案。temp数组记录每一次乘积之后的内容。数组a为输入的矩阵，需要乘m次a。

#include <bits/stdc++.h>

using namespace std;

int ans[33][33],temp[33][33],a[33][33];

int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        cin>>a[i][j];

    for(int i=1;i<=n;i++)//行
            for(int j=1;j<=n;j++)//列
            ans[i][j]=a[i][j];
    if(m)
    {
        m--;
        while(m)//m次幂
        {
            m--;
            memset(temp,0,sizeof(temp));
            for(int i=1;i<=n;i++)//temp的行
                for(int j=1;j<=n;j++)//temp的列
                    for(int k=1;k<=n;k++)//
                        temp[i][j]+=ans[i][k]*a[k][j];
            for(int i=1;i<=n;i++)//temp的行
                for(int j=1;j<=n;j++)//temp的列
                ans[i][j]=temp[i][j];
        }
    }
    else
    {
        memset(ans,0,sizeof(ans));
        for(int i=1;i<=n;i++)
            ans[i][i]=1;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            cout<<ans[i][j];
            if(j==n)cout<<endl;
            else cout<<" ";
        }
    return 0;
}