天梯赛 L1-048 矩阵A乘以B (15 分) 简单循环而已

天梯赛 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

简单分析：

就是一个三重循环的问题，要搞清楚的是因为ca=rb，所以才只需要三重循环。

代码

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 110;

int ra, ca, rb, cb;
int A[N][N], B[N][N], C[N][N];

int main() {
	cin >> ra >> ca;
	for (int i = 0; i < ra; i++)
		for (int j = 0; j < ca; j++)
			cin >> A[i][j];
	cin >> rb >> cb;
	for (int i = 0; i < rb; i++)
		for (int j = 0; j < cb; j++)
			cin >> B[i][j];
	if (ca != rb) {
		cout << "Error: " << ca << " != " << rb << endl;
	}
	else {
		for (int i = 0; i < ra; i++)
			for (int j = 0; j < cb; j++)
				for (int k = 0; k < rb; k++)
					C[i][j] += A[i][k] * B[k][j];

		cout << ra << ' ' << cb << endl;
		for (int i = 0; i < ra; i++) {
			for (int j = 0; j < cb; j++) {
				cout << C[i][j];
				if (j != cb - 1) cout << ' ';
			}
			cout << endl;
		}
	}

	return 0;
}

总结感悟

简单语法题，套了一层矩阵的壳别吓到了~ ~