蓝桥杯 基础练习 矩阵乘法

问题描述
　　给定一个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 <iostream>
using namespace std;
 
long long num[30][30]={0};   
long long num1[30][30]={0};
long long num2[30][30]={0};
 
int main()
{
	int n,m;      //n阶 m次方 
	int i,j,z;
	cin >> n >> m;
	for(i=0;i<n;i++)
	{
		for(j=0;j<n;j++)
		{
			cin >> num[i][j];
			num1[i][j]= num[i][j];   //对num1赋值，用于相乘 
		}
	}
	if(m==0)   //等于0的时候 
	{ 
		for(i=0;i<n;i++)
		{	
			num2[i][i] = 1;
		}	
	}
	else if(m==1)  //等于1的时候 
	{
		for(i=0;i<n;i++)
			for(j=0;j<n;j++)
				num2[i][j] = num[i][j]; 
	}
	else  //m大于1的时候 
	{
		for(int count=1;count<m;count++)  //次数 
		{
			for(i=0;i<n;i++)         
			{
				for(j=0;j<n;j++)
				{
					num2[i][j]=0;
					for(z=0;z<n;z++)
					{
						num2[i][j] += num[i][z] * num1[z][j];   //num横向,num1纵向 
					}
				}
			}
			for(int x=0;x<n;x++)
				for(int y=0;y<n;y++)
					num[x][y] = num2[x][y];
		}
	}
	for(i=0;i<n;i++)
	{
		for(j=0;j<n;j++)
			cout << num2[i][j] << " ";
		cout << endl;
	}
	return 0;
}