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

注意事项: 恶补一波线性代数，矩阵的零次幂竟然是单位矩阵，太伤心了┭┮﹏┭┮

#include<iostream>
#include<algorithm>
#include<cmath>
#include<string>
#include<cstring>
using namespace std;
int a[1001][1001];
int b[1001][1001];
int c[1001][1001];
int main()
{
	int n, m;
	cin >> n >> m;
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			cin >> a[i][j];
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			b[i][j] = a[i][j];
	if (m == 0)
	{
		for (int i = 0; i < n; i++)
			for (int j = 0; j < n; j++)
			{
				if (i == j)
					a[i][j] = 1;
				else
					a[i][j] = 0;
			}
	}
	else
	{
		for (int i = 1; i < m; i++)
		{
			memset(c, 0, sizeof(c));
			for (int j = 0; j < n; j++)
			{
				for (int k = 0; k < n; k++)
				{
					for (int l = 0; l < n; l++)
					{
						c[j][k] += a[j][l] * b[l][k];
					}
				}
			}
			for (int j = 0; j < n; j++)
				for (int k = 0; k < n; k++)
					a[j][k] = c[j][k];
		}
	}
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			cout << a[i][j];
			if (j != n)
				cout << " ";
		}
		cout << endl;
	}
	return 0;
}