1472: 蓝桥杯基础练习VIP-矩阵乘法

解题思路:

注意事项:

参考代码://
// Created by ch'li'sen on 2022/4/19.
//

#include "iostream"
#include "string.h"
using namespace std;
int n,m;
int a[30][30],b[30][30],c[30][30];
void f(){
   memset(c,0,sizeof(c));
   for (int i = 0; i < n; ++i) {
       for (int j = 0; j < n; ++j) {
           for (int k = 0; k < n; ++k) {
              c[i][j]+=a[i][k]*b[k][j];
           }
       }
   }
   for (int i = 0; i < n; ++i) {
       for (int j = 0; j < n; ++j) {
           a[i][j]=c[i][j];
       }
   }
}
int main()
{
   cin>>n>>m;
   for (int i = 0; i < n; ++i) {
       for (int j = 0; j < n; ++j) {
           cin>>a[i][j];
           b[i][j]=a[i][j];
       }
   }
   if(m==0){
       memset(a,0,sizeof (a));
       for (int i = 0; i < n; ++i) {
           a[i][i]=1;
       }
   }
   else if(m==1){
   }else{
       for(int i=1;i<m;i++){
           f();
       }
   }
   for (int i = 0; i < n; ++i) {
       for (int j = 0; j < n; ++j) {
           cout<<a[i][j]<<" ";
       }
       cout<<"\n";
   }
   return 0;
}