蓝桥杯--矩阵乘法

http://47.104.209.207/problem/old1059

直接用矩阵乘法的性质，注意0次方的时候是变成单位矩阵。

#include<cmath>
#include<iostream>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
const int N=40;
int matrix1[N][N],matrix2[N][N];
int t[N][N];
int n,m;
void mul(){
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            int sum=0;
            for(int k=0;k<n;k++){
                sum+=matrix1[i][k]*matrix2[k][j];
            }
            t[i][j]=sum;
        }
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            matrix2[i][j]=t[i][j];
        }
    }
}
int main(){
    cin>>n>>m;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            cin>>matrix1[i][j];
            matrix2[i][j]=matrix1[i][j];
        }
    }
    for(int i=0;i<m-1;i++){
        mul();
    }
    if(m>0){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                cout<<matrix2[i][j]<<" ";
            }
            cout<<endl;
        }
    }else{
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(i==j) cout<<"1"<<" ";
                else cout<<"0"<<" ";
            }
            cout<<endl;
        }
    }
    return 0;
}