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 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; }
|