蓝桥杯题目练习(矩阵乘法)

基础练习VIP 矩阵乘法

题目描述
给定一个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 <iomanip>
#include <algorithm>
#include <cmath>
using namespace std;

int main()
{
	int n, m;
	cin >> n >> m;
	int matrix[30][30];
	int rmatrix[30][30];
	int tempmatrix[30][30];
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			cin >> matrix[i][j];
			tempmatrix[i][j] = matrix[i][j];
		}
	}
	if (m == 1)
	{
		cout << pow(matrix[0][0], m);
	}
	else if (m == 0)
	{
		for (int i = 0; i < n; i++)
		{
			for (int j = 0; j < n; j++)
			{
				if (i == j)
					cout << 1 << " ";
				else
					cout << 0 << " ";
			}
			cout << endl;
		}
	}
	else
	{
		for (int count = 1; count < m; count++)
		{
			for (int i = 0; i < n; i++)
			{
				for (int j = 0; j < n; j++)
				{
					rmatrix[i][j] = 0;
					for (int k = 0; k < n; k++)
					{
						rmatrix[i][j] += tempmatrix[i][k] * matrix[k][j];
					}
				}
			}
			for (int i = 0; i < n; i++)
			{
				for (int j = 0; j < n; j++)
				{
					tempmatrix[i][j] = rmatrix[i][j];
				}
			}
		}
		for (int i = 0; i < n; i++)
		{
			for (int j = 0; j < n; j++)
			{
				cout << rmatrix[i][j] << " ";
			}
			cout << endl;
		}
	}


	
	return 0;
}