蓝桥杯——矩阵相乘

问题描述
  给定一个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

解题思路：首先要了解矩阵乘法运算规则：

1、用A的第1行各个数与B的第1列各个数对应相乘后加起来,就是乘法结果中第1行第1列的数（对应相乘，即第一个数乘第一个，第二个数乘第二个，然后相加）；

2、用A的第1行各个数与B的第2列各个数对应相乘后加起来,就是乘法结果中第1行第2列的数；

3、用A的第1行各个数与B的第3列各个数对应相乘后加起来,就是乘法结果中第1行第3列的数；

依次进行，（直到）用A的第1行各个数与B的第末列各个数对应相乘后加起来,就是乘法结果中第1行第末列的的数。

二：由于是求矩阵的多少次幂， 那么首先可以想到要有一个二维数组用来储存矩阵本身，再用第二个数组来表示每次相乘后变化的结果。下面上代码：

Code:

#include <iostream>
#include <stdio.h>

using namespace std;

const int N = 103;

int a[N][N], b[N][N], c[N][N];
int n, m, l;

signed main()
{
	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) //特判，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;
		}
	}
	else if(m == 1) //1次幂即本身
	{
		for(int i = 1; i <= n; i ++ )
		{
			for(int j = 1; j <= n; j ++ )
			{
				cout << b[i][j] << " ";
			}
			cout << endl;
		}
	}
	else
	{
		for(int o = 1; o < m; o ++ )
		{
			for(int i = 1; i <= n; i ++ )
			{
				for(int j = 1; j <= n; j ++ )
				{
					l = 0;
					for(int k = 1; k <= n; k ++ )
					{
						l += b[i][k] * a[k][j]; //对应的各个元素相乘
					}
					c[i][j] = l;
				}
			}
			for(int i = 1; i <= n; i ++ )
			{
				for(int j = 1; j <= n; j ++ )
				{
					b[i][j] = c[i][j]; //这是重点，b[][]每次循环都在发生变化
				}
			}
		}
		for(int i = 1; i <= n; i ++)
		{
			for(int j = 1; j <= n; j ++ )
			{
				cout << b[i][j] << " ";
			}
			cout << endl;
		}
	}
}