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的行数

输入样例

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

输出样例

2 4
20 22 24 16
53 58 63 28

解题思路

源码

#include <iostream>
using namespace std;

int main() {
    int a,b,c,d;
    cin >> a >> b;
    int matrix1[a][b];
    for (int i = 0;i < a;i++) {
        for (int j = 0;j < b;j++) {
            cin >> matrix1[i][j];
        }
    }
    cin >> c >> d;
    int matrix2[c][d];
    for (int i = 0;i < c;i++) {
        for (int j = 0; j < d;j++) {
            cin >> matrix2[i][j];
        }
    }
//  int matrix3[a][d];
    if (b != c) {
        cout << "Error: "<<b <<" != " <<c;
    } else {
        cout << a << " " << d << endl;
        int countCol = 0;
        for (int i = 0;i < a;i++) {
            for (int j = 0;j < d;j++) {
                int sum = 0;
                 for (int k = 0;k < b;k++){
                    sum = sum + matrix1[i][k]*matrix2[k][j];
                 }
                 if (j < d-1) {
                    cout << sum << " ";
                 } else {
                    cout << sum << endl; 
                 }
            }
        }
    }
    return 0;
}