【PTA】​L1-048 矩阵A乘以B​ (C++)

简介: 【PTA】​L1-048 矩阵A乘以B​ (C++)

题目要求:

给定两个矩阵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 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;
}

测试结果:


目录
相关文章
|
2月前
|
算法 测试技术 C++
【动态规划】【矩阵快速幂】【滚动向量】C++算法552. 学生出勤记录 II
【动态规划】【矩阵快速幂】【滚动向量】C++算法552. 学生出勤记录 II
|
2月前
|
C++
【PTA】L1-016 验证身份(C++)
【PTA】L1-016 验证身份(C++)
39 0
【PTA】L1-016 验证身份(C++)
|
3天前
|
人工智能 小程序 BI
矩阵的转置、加和乘法写入C++
矩阵的转置、加和乘法写入C++
6 0
|
18天前
|
算法 测试技术 Serverless
【二分查找】【C++算法】378. 有序矩阵中第 K 小的元素
【二分查找】【C++算法】378. 有序矩阵中第 K 小的元素
|
2月前
|
算法 测试技术 C++
【字符串】【 LCP】【C++算法】2573找出对应 LCP 矩阵的字符串
【字符串】【 LCP】【C++算法】2573找出对应 LCP 矩阵的字符串
|
2月前
|
算法 C++
【动态规划】【矩阵】C++算法329矩阵中的最长递增路径
【动态规划】【矩阵】C++算法329矩阵中的最长递增路径
|
2月前
|
C++
【PTA】L1-011 A-B (C++)
【PTA】L1-011 A-B (C++)
60 0
【PTA】L1-011 A-B (C++)
|
2月前
|
C++
【PTA】​L1-005 考试座位号​ (C++)
【PTA】​L1-005 考试座位号​ (C++)
68 0
【PTA】​L1-005 考试座位号​ (C++)
|
2月前
|
测试技术 C++
【PTA】​L1-003 个位数统计​ (C++)
【PTA】​L1-003 个位数统计​ (C++)
44 0
【PTA】​L1-003 个位数统计​ (C++)
|
2月前
|
C++
【PTA】L1-020 帅到没朋友 (C++)
【PTA】L1-020 帅到没朋友 (C++)
56 0
【PTA】L1-020 帅到没朋友 (C++)