蓝桥杯-基础练习-VIP试题-矩阵乘法（BASIC-17）

问题描述

　　给定一个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>
#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;
	}
//	cout<<endl;
return 0;
	
	
}