C++实现矩阵乘法

题目描述
给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
例如：
A =
1 2
3 4
A的2次幂
7 10
15 22

输入
第一行是一个正整数N、M（1< =N< =30, 0< =M< =5），表示矩阵A的阶数和要求的幂数
接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值

输出
输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开

样例输入
2 2
1 2
3 4

样例输出
7 10
15 22

C++实现代码如下:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
int a[40][40];
int c[40][40];   //存储每次相乘的结果
int b[40][40];   //传递每次相乘的结果，到下一次乘法中
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    if(m==0){   //0次幂输出单位矩阵，对角线为1，其他位0
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(i==j){
                    b[i][j]=1;   //至于为什么是b，可以先看后面
                }else{
                    b[i][j]=0;
                }
            }
        }        
    }else if(m!=1){
        m=m-1;   //注意：两次幂只做一次乘法，三次幂只做两次乘法
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                scanf("%d",&a[i][j]);
                b[i][j]=a[i][j];    //输入矩阵a，并复制到矩阵b
            }
        }
        while(m--){    //m次循环
            for(int i=0;i<n;i++){
                for(int j=0;j<n;j++){   //i，j确定
                    for(int t=0;t<n;t++){
                        c[i][j]+=b[i][t]*a[t][j]; //例如：c[3][4]就是a[3][0~t]*b[0~t][4]累加
                    }
                }
            }
            for(int i=0;i<n;i++){  //复制c到b，b作为这次乘法的结果又拿回去跟a相乘
                for(int j=0;j<n;j++){
                    b[i][j]=c[i][j];
                    c[i][j]=0;   //c要重新初始化
                }
            }
        }        
    }
    //输出结果
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            cout<<b[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}