L1-048 矩阵A乘以B

最新推荐文章于 2023-07-08 18:11:06 发布

Zero_979

最新推荐文章于 2023-07-08 18:11:06 发布

阅读量242

点赞数

分类专栏： PTA 文章标签：程序设计 c++ 算法

本文链接：https://blog.csdn.net/Zero_979/article/details/106076454

版权

PTA 专栏收录该内容

54 篇文章 4 订阅

订阅专栏

原题地址：PTA-团体程序设计天梯赛-练习集

题目：

给定两个矩阵A和B，要求你计算它们的乘积矩阵AB。需要注意的是，只有规模匹配的矩阵才可以相乘。即若A有Ra行、Ca列，B有Rb行、Cb列，则只有Ca与Rb相等时，两个矩阵才能相乘。

输入格式：

输入先后给出两个矩阵A和B。对于每个矩阵，首先在一行中给出其行数R和列数C，随后R行，每行给出C个整数，以1个空格分隔，且行首尾没有多余的空格。输入保证两个矩阵的R和C都是正数，并且所有整数的绝对值不超过100。

输出格式：

若输入的两个矩阵的规模是匹配的，则按照输入的格式输出乘积矩阵AB，否则输出Error: Ca != Rb，其中Ca是A的列数，Rb是B的行数。

输入样例1：

2 3
1 2 3
4 5 6
3 4
7 8 9 0
-1 -2 -3 -4
5 6 7 8

输出样例1：

2 4
20 22 24 16
53 58 63 28

输入样例2：

输出样例2：

Error: 2 != 3

分析：

以前大一的时候不会矩阵，所以现在来补坑。。

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int main()
{
	int a[100][100];
	int b[100][100];
	int r1, l1, r2, l2;
	while (cin >> r1 >> l1) {
		memset(a, 0, sizeof(a));
		memset(b, 0, sizeof(b));
		for (int i = 0; i < r1; i++)
			for (int j = 0; j < l1; j++)
				cin >> a[i][j];
		cin >> r2 >> l2;
		for (int i = 0; i < r2; i++)
			for (int j = 0; j < l2; j++)
				cin >> b[i][j];
		if (l1 != r2)printf("Error: %d != %d\n", l1, r2);
		else {
			int c[100][100];
			memset(c, 0, sizeof(c));
			for (int i = 0; i < r1; i++)
			{
				for (int j = 0; j < l2; j++)
				{
					for (int k = 0; k < l1; k++)
					{
						c[i][j] += a[i][k] * b[k][j];
						//cout << a[i][k]<<" "<<b[k][j] << endl;
					}
				}
			}
			cout << r1 << " " << l2 << endl;
			for (int i = 0; i < r1; i++) {
				for (int j = 0; j < l2; j++) {
					if (j != 0)cout << " ";
					cout << c[i][j];
				}
				cout << endl;
			}
		}
	}
	return 0;
}