试题 基础练习 矩阵乘法

写在前面： 一开始理解不了矩阵相乘怎么实现，看了别人的写法才懂了，然后就是存储数据输出的问题，还有就是边界情况要考虑好

矩阵相乘模板（两个及以上） M为幂次，N为矩阵的秩，这个模板理解不了的可以先记下来

for(int l=0;l<M-1;l++){
			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++)
						b[i][j]=c[i][j];	
			memset(c,0,sizeof(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

完整代码

#include 
using namespace std;	
int main()
{
	int a[100][100],b[100][100],c[100][100];
	//初始化 
	memset(a,0,sizeof(a));
	memset(b,0,sizeof(b));
	memset(c,0,sizeof(c));
	int M,N;
	cin>>N>>M;
	for(int i=0;i<N;i++){
		for(int j=0;j<N;j++){
			c[i][j]=0;
			cin>>a[i][j];
			b[i][j]=a[i][j];//复制一下 
		}
	}		
	//矩阵为单位 
	if(M==0){
		for(int i=0;i<N;i++){
			for(int j=0;j<N;j++){
				if(i==j)
					b[i][j]=1;
				else
					b[i][j]=0;
			}
		}
	}
	//转置一次为本身 
	else if(M==1){
		for(int i=0;i<N;i++)
			for(int j=0;j<N;j++)
				b[i][j]=a[i][j];	
	}
	//矩阵相乘	
	else{
		for(int l=0;l<M-1;l++){
			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++)
						b[i][j]=c[i][j];	
			memset(c,0,sizeof(c));	//跑完一次要重置，用于新一轮继续累加 
		}
	}
	//输出答案	
	for(int i=0;i<N;i++){
		for(int j=0;j<N;j++){
			cout<<b[i][j]<<" ";	
		}
		cout<<endl;			
	}	
	return 0;
}