蓝桥杯 BASIC_17 矩阵乘法 （矩阵快速幂）

#include<iostream>
#include<string.h>
using namespace std;
struct M{
    int num[40][40];
    M(){
        memset(num,0,sizeof(num));
    }
};
M a,e;
int m;
M mul(M a,M b){//计算矩阵乘法
    M c;
    for(int i=0;i<m;i++){
        for(int j=0;j<m;j++){
            for(int k=0;k<m;k++){
                c.num[i][j]+=(a.num[i][k]*b.num[k][j]);
            }
        }
    }
    return c;
}
M multi(M c,int n){//矩阵快速冥核心代码
    M b=c,r=e;
    while(n){
        if(n&1){
            r=mul(r,b);
        }
        b=mul(b,b);
        n>>=1;
    }
    return r;
}
int main(){
    int n;
    cin>>m>>n;
    for(int i=0;i<m;i++){
        e.num[i][i]=1;
        for(int j=0;j<m;j++){
            cin>>a.num[i][j];
        }
    }
    M x = multi(a,n);
    for(int i=0;i<m;i++){
        for(int j=0;j<m;j++){
            cout<<x.num[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}