ZCMU-2045: 矩阵乘法

2045: 矩阵乘法

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 105   Solved: 28
[ Submit][ Status][ Web Board]

Description

给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
例如：
A =
1 2
3 4
A的2次幂
7 10
15 22

Input

第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数
接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值

Output

输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。每个数后输出一个空格。

Sample Input

2 2

1 2

3 4

Sample Output

7 10

15 22

HINT

Source

【解析】

线性代数就是烦，但是 也很考验一个人思维是否谨慎，像我就不行，还有很长的路要走。

得有线性代换的基础再来写这题。

#include <bits/stdc++.h>
using namespace std;
int c[35][35], a[35][35],b[35][35];
void fun(int n, int m)//这里用很简单的递归写。n就是阶数，只有方阵才有幂。m是幂数
{
	if (1 == m)return;//判断边界
	for (int i = 1; i <= n; i++) //行
	{
		for (int j = 1; j <= n; j++) //列
		{
			int t = 0;					//弄一个临时变量存结果矩阵c[i][j]的值
			for (int k = 1; k <= n; k++) //矩阵中的每一个数
			{
				t += b[i][k] * a[k][j]; //为什么要引入b矩阵？
										//因为b是上一次幂运算结束时候的c，不引入的话 自己调试一下就发现会错了
			}
			c[i][j] = t;
		}
	}
	memcpy(b, c, sizeof(c));//把这次运算完的结果c存到b里去
	fun(n, m - 1);          //进入递归
}
int main()
{
	int n, m;
	while (~scanf("%d%d", &n, &m))
	{
		for (int i = 1; i <= n; i++)
			for (int j = 1; j <= n; j++)
			{
				scanf("%d", &a[i][j]);
				b[i][j] = a[i][j];
			}
		/*if (m > 0)
		{
			fun(n, m);
			for (int i = 1; i <= n; i++)
			{
				for (int j = 1; j <= n; j++)
				{
					printf("%d ", c[i][j]);
				}
				printf("\n");
			}
		}
		else if (m == 0)
		{
			for (int i = 1; i <= n; i++)
			{
				for (int j = 1; j <= n; j++)
				{
					if (i == j)printf("1 ");
					else printf("0 ");
				}
				printf("\n");
			}
		}*/
		fun(n, m);
		/*这是我看来的很骚的操作，但原理是一样的，就是把我上面的一大段输出写的骚气。一点喜欢*/
		for (int i = 1; i <= n; i++)
		{
			for (int j = 1; j <= n; j++)
				printf("%d ", m == 0 ? (i == j ? 1 : 0) : c[i][j]);
			printf("\n");
		}
	}
	return 0;
}