蓝桥杯基础练习VIP-矩阵乘法-题解（C++代码）

参考代码:

#include<iostream>
using namespace std;

int **Init_arr(int n) //初始化二维数组
{
	int **arr;
	arr = new int*[n];
	for (int i = 0;i < n;i++)
		arr[i] = new int[n];
	return arr;
}
void arr(int **a, int **b, int **c, int n, int m)
{
	if (m == 0) {  //矩阵的幂为0就输出单位矩阵
		for (int i = 0;i < n;i++) {
			for (int j = 0;j < n;j++)
				i == j ? b[i][j] = 1 : b[i][j] = 0;
		}
		return;
	}
	for (int x = 1;x < m;x++) { //控制幂的次数
		for (int i = 0;i < n;i++)
			for (int j = 0;j < n;j++) {
				c[i][j] = 0;
				for (int k = 0;k < n;k++) {
					c[i][j] += a[i][k] * b[k][j]; //a的列等于b的行，a的每一行乘以b的每一列
				}
			}
		for (int i = 0;i < n;i++)
			for (int j = 0;j < n;j++)
				b[i][j] = c[i][j];
	}
}
int main()
{
	int n, m;
	cin >> n >> m;
	int **a, **b, **c;
	a = Init_arr(n);
	b = Init_arr(n);
	c = Init_arr(n);
	for (int i = 0;i < n;i++)
		for (int j = 0;j < n;j++) {
			cin >> a[i][j];
			b[i][j] = a[i][j];
		}
	arr(a, b, c, n, m);
	for (int i = 0;i < n;i++) {
		for (int j = 0;j < n;j++)
			cout << b[i][j] << " ";
		cout << endl;
	}
	return 0;
}