蓝桥杯练习系统 BASIC-17 VIP试题 矩阵乘法（c++）

试题 基础练习 矩阵乘法

资源限制

时间限制：1.0s 内存限制：512.0MB

问题描述

　　给定一个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<bits/stdc++.h>
using namespace std;

int n,m;
int a[50][50];
int b[50][50];
int c[50][50];

int main(){
	int sum;
	cin>>n>>m;
	
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>a[i][j];
			b[i][j]=a[i][j];
		}
	}
	if(m==0){
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				if(i==j)cout<<'1'<<' ';
				else cout<<'0'<<' '; 
			}
			cout<<endl;
		}
		return 0;
	}
	m--;
	while(m--){
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				sum=0;
				for(int k=1;k<=n;k++){		
					sum+=(b[i][k]*a[k][j]);
				//	cout<<sum<<"+="<<b[i][k]<<'*'<<a[k][j]<<endl;
				}
				c[i][j]=sum;
			//	cout<<i<<' '<<j<<':'<<sum<<endl;
			}
			
		}
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				b[i][j]=c[i][j];
			}
		//	cout<<endl;
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cout<<c[i][j]<<' ';
		}
		cout<<endl;
	}
	return 0;
}