基础练习 矩阵乘法

试题 基础练习 矩阵乘法(蓝桥杯，详细注释)

资源限制
时间限制：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

分析：

首先我们要知道矩阵的乘法计算：
假设有两个矩阵 a,b— m,n
----------------------- c,d — s,k
那么相乘的矩阵为： am+bs ------ an+bk
----------------------------- cm+ds ------ cn+dk
还有就是单位矩阵，当m为0时。我们分三种情况讨论。
1：单位矩阵
2:本身矩阵
3：其他情况

附上代码：

#include<iostream>
using namespace std;
int main()
{
	int a[35][35] = { 0}; int n, m;//n表示多大矩阵，m表示相乘几次
	int c[35][35] = {0}, b[35][35] = { 0 };
	cin >> n >> m;
	int i, j;
	for (i = 1; i <= n; i++)
		for (j = 1; j <= n; j++)
		{
			cin >> a[i][j];
			b[i][j] = a[i][j];//b数组和a数组一样
		}
	if (m == 0)//输出为单位矩阵
	{
		for (i = 1; i <= n; i++)
		{
			for (j = 1; j <= n; j++)
			{
				if (i == j)
					cout << 1 << " ";
				else
					cout << 0 << " ";
			}
			cout << endl;//输出一行后换行
		}
	}
	if (m == 1)//输出为本身矩阵
	{
		for (i = 1; i <= n; i++)
		{
			for (j = 1; j <= n; j++)
			{
				cout << a[i][j];
			}
			cout <<endl;
		}
	}
	if (m >= 2)//开始相乘的矩阵
	{
		int k;//v代表每一个求和的数然后赋给对应的位置
		for (k = 2; k <= m; k++)//执行相乘的次数
		{
			for (i = 1; i <= n; i++)
			{
				for (j = 1; j <= n; j++)
				{int  v = 0;
					for (int z = 1; z <= n; z++)
					{
						v = v + a[i][z] * b[z][j];
					}
					c[i][j] = v;//存放加和结果
				}
			}
			for (i = 1; i <= n; i++)
			{
				for (j = 1; j <= n; j++)
				{
					a[i][j] = c[i][j];//让c[][]赋给a[][],再让a[][]*b[][]
				}
			}
		}

		for (i = 1; i <= n; i++)
		{
			for (j = 1; j <=n; j++)
			{
				cout<<a[i][j]<<" ";
			}
			cout << endl;
		}

	}

	return 0;
}

运行代码：