團體程序設計天梯賽-練習集-L1-048. 矩陣A乘以B

L1-048. 矩陣A乘以B

給定兩個矩陣A和B,要求你計算它們的乘積矩陣AB。須要注意的是,只有規模匹配的矩陣才能夠相乘。即若A有Ra行、Ca列,B有Rb行、Cb列,則只有Ca與Rb相等時,兩個矩陣才能相乘。c++

輸入格式:spa

輸入前後給出兩個矩陣A和B。對於每一個矩陣,首先在一行中給出其行數R和列數C,隨後R行,每行給出C個整數,以1個空格分隔,且行首尾沒有多餘的空格。輸入保證兩個矩陣的R和C都是正數,而且全部整數的絕對值不超過100。code

輸出格式:blog

若輸入的兩個矩陣的規模是匹配的,則按照輸入的格式輸出乘積矩陣AB,不然輸出「Error: Ca != Rb」,其中Ca是A的列數,Rb是B的行數。ci

輸入樣例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
作法:兩矩陣相乘,左矩陣第一行乘以右矩陣第一列(分別相乘,第一個數乘第一個數),乘完以後相加,即爲結果的第一行第一列的數
 1 #include<bits/stdc++.h>
 2 
 3 using namespace std;
 4 
 5 int a[110][110], b[110][110], c[110][110], ra, ca, rb, cb;
 6 
 7 int main() {
 8     cin >> ra >> ca;
 9     for (int i = 0; i < ra; i++) {
10         for (int j = 0; j < ca; j++) {
11             cin >> a[i][j];
12         }
13     }
14     cin >> rb >> cb;
15     for (int i = 0; i < rb; i++) {
16         for (int j = 0; j < cb; j++) {
17             cin >> b[i][j];
18         }
19     }
20     if (ca == rb) {
21         for (int i = 0; i < ra; i++) {
22             for (int j = 0; j < cb; j++) {
23                 c[i][j] = 0;
24                 for (int k = 0; k < rb; k++) {
25                     c[i][j] += (a[i][k] * b[k][j]);
26                 }
27             }
28         }
29         printf ("%d %d\n", ra, cb);
30         for (int i = 0; i < ra; i++) {
31             for (int j = 0; j < cb; j++) {
32                 printf ("%d%c", c[i][j],(j == cb - 1) ? '\n' : ' ');
33             }
34         }
35     } else
36         printf ("Error: %d != %d\n", ca, rb);
37     return 0;
38 }
相關文章
相關標籤/搜索