?題目鏈接:L1-048 矩陣A乘以B
題目要求:
?給定兩個矩陣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:
3 2
38 26
43 -5
0 17
3 2
-11 57
99 68
81 72
思路:
1.先將兩個數據分別用for循環的方式遍歷到a,b數組中
2.如果ca與rb不相同,也就是a數組的列數與b數組的行數不相同時則打印錯誤原因
3.如果ca與rb相同,則遍歷a數組的行數ra,b數組的列數cb,a,b數組相同的ca與rb,進行乘積相合
4.最后輸出合并后的新行數和列數,也就是ra和cb
代碼:
#include <bits/stdc++.h>using namespace std;#define N 1000int main()
{int ra,ca,rb,cb;int a[N][N],b[N][N];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)printf("Error: %d != %d\n",ca,rb);else{cout << ra << " " << cb << endl;for(int i = 0; i < ra; i ++){for(int j =0; j < cb; j ++){int sum = 0;for(int k = 0; k < ca; k ++){sum += a[i][k] * b[k][j];}cout << sum ;if(j != cb - 1){cout << " ";}if(j == cb - 1){cout << endl;}}}}return 0;
}