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

测试结果:


目录
相关文章
|
8月前
|
算法 测试技术 C++
【动态规划】【矩阵快速幂】【滚动向量】C++算法552. 学生出勤记录 II
【动态规划】【矩阵快速幂】【滚动向量】C++算法552. 学生出勤记录 II
|
8月前
|
C++
【PTA】L1-016 验证身份(C++)
【PTA】L1-016 验证身份(C++)
101 0
【PTA】L1-016 验证身份(C++)
|
7月前
|
存储 C++
【PTA】L1-039 古风排版(C++)
【PTA】L1-039 古风排版(C++)
61 1
|
8月前
|
Java C++
部落(pta)(并查集) Java以及C++
部落(pta)(并查集) Java以及C++
67 2
|
5月前
|
C++
C++ PCL 计算多个RT矩阵变换后的变换矩阵
C++ PCL 计算多个RT矩阵变换后的变换矩阵
61 0
|
7月前
|
C++
C++解决线性代数矩阵转置 小实践
【6月更文挑战第3天】C++解决线性代数矩阵转置
101 2
|
7月前
|
存储 人工智能 C++
【PTA】L1-064 估值一亿的AI核心代码(详C++)
【PTA】L1-064 估值一亿的AI核心代码(详C++)
64 1
|
7月前
|
存储 C++ 索引
【PTA】L1-059 敲笨钟(C++)
【PTA】L1-059 敲笨钟(C++)
56 1
|
7月前
|
存储 人工智能 C++
【PTA】L1-093 猜帽子游戏(C++)
【PTA】L1-093 猜帽子游戏(C++)
141 1
|
7月前
|
C++
【PTA】L1-046 整除光棍(C++)
【PTA】L1-046 整除光棍(C++)
77 1