【C++】蓝桥基础练习14 矩阵乘法

最新推荐文章于 2024-08-14 17:18:35 发布

Rluhan7

最新推荐文章于 2024-08-14 17:18:35 发布

阅读量301

点赞数 6

分类专栏：蓝桥练习文章标签： c++ 矩阵算法

本文链接：https://blog.csdn.net/Rluhan7/article/details/137434652

版权

蓝桥练习专栏收录该内容

16 篇文章 0 订阅

订阅专栏

本文介绍了如何使用C++编程语言计算给定矩阵的幂次，并提供了代码示例，同时讨论了时间限制和内存管理。算法利用了迭代方式，避免了重复计算，适用于N阶矩阵和较小的幂次M。

摘要由CSDN通过智能技术生成

资源限制

内存限制：512.0MB C/C++时间限制：1.0s Java时间限制：3.0s Python时间限制：5.0s

问题描述

　　给定一个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

代码注意数组更新保证运算能再次进行

#include <iostream>
using namespace std;

#define N 30
#define M 5
int a[N][N],b[N][N];
int ans[N][N];

void f(int n) {
	
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			int sum = 0;
			for (int k = 0; k < n; k++) {
				sum += b[i][k] * a[k][j];
			}
			ans[i][j] = sum;
		}
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			b[i][j] = ans[i][j];//保证下一次运算的进行
		}
	}
}

int main() {
	int m, n;
	cin >> n >> m;
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			cin >> a[i][j];
			if (m == 0) {
				if (i == j) {
					ans[i][j] = 1;
				}
				else {
					ans[i][j] = 0;
				}
			}
			else if (m == 1) {
				ans[i][j] = a[i][j];
			}
			else {
				b[i][j] = a[i][j];
			}
		}
	}
	for (int i = 2; i <= m; i++) {
		f(n);
	}
	
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			cout<< ans[i][j]<<" ";
		}
		cout << endl;
	}
	return 0;
}