每日一练2023.12.9—— 矩阵A乘以B【PTA】

news2024/11/15 22:29:24

题目链接：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：

思路：

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 1000

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