L1-048 矩阵A乘以B (15 分)

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

输入格式:
输入先后给出两个矩阵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:
3 2
38 26
43 -5
0 17
3 2
-11 57
99 68
81 72
输出样例2:
Error: 2 != 3

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<string>
#include<iostream>
#include<algorithm>
#include<map>
#include<stack>
#include<set>
#include<vector>
#include<queue>
#define ll long long
#define dd double
using namespace std;

ll matrix_A[1050][1050];
ll matrix_B[1050][1050];

int main() {
	ll matrix_A_line;
	ll matrix_A_list;
	ll matrix_B_line;
	ll matrix_B_list;
	ll sum;
	cin >> matrix_A_line >> matrix_A_list;
	for (ll i = 0; i < matrix_A_line; i++) {
		for (ll j = 0; j < matrix_A_list; j++) {
			cin >> matrix_A[i][j];
		}
	}
	cin >> matrix_B_line >> matrix_B_list;
	for (ll i = 0; i < matrix_B_line; i++) {
		for (ll j = 0; j < matrix_B_list; j++) {
			cin >> matrix_B[i][j];
		}
	}
	if (matrix_A_list != matrix_B_line) {
		cout << "Error: " << matrix_A_list << " != " << matrix_B_line << endl;
	}
	else {
		cout << matrix_A_line << " " << matrix_B_list << endl;
		for (ll i = 0; i < matrix_A_line; i++) {
			for (ll j = 0; j < matrix_B_list; j++) {
				sum = 0;
				for (ll k = 0; k < matrix_A_list; k++) {
					sum += matrix_A[i][k] * matrix_B[k][j];
				}
				if (j != 0) {
					cout << " ";
				}
				cout << sum;
			}
			cout << endl;
		}
	}
}

猜你喜欢

转载自blog.csdn.net/weixin_44231195/article/details/90030606