L1-048 矩阵A乘以B(模拟)

给定两个矩阵 $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:

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

输出样例1:

1
2
3
2 4
20 22 24 16
53 58 63 28

输入样例2:

1
2
3
4
5
6
7
8
3 2
38 26
43 -5
0 17
3 2
-11 57
99 68
81 72

输出样例2:

1
Error: 2 != 3

思路:

矩阵相乘,用二维数组模拟即可。

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#include <bits/stdc++.h>
using namespace std;

int ra, ca, rb, cb, sum = 0;
int a[105][105], b[105][105];
string s;

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 << ra << " " << cb << endl;
for(int i = 0; i < ra; i++)
{
for(int j = 0; j < cb; j++)
{
for(int k = 0; k < rb; k++)
{
sum += a[i][k] * b[k][j];
}
if(j)
cout << " " << sum;
else
cout << sum;
sum = 0;
}
cout << endl;
}
}
else
cout << "Error: " << ca << " != " << rb;

return 0;
}