PTA L1-048 矩阵A乘以B

最新推荐文章于 2024-05-03 16:18:45 发布

Artiel

最新推荐文章于 2024-05-03 16:18:45 发布

阅读量3.3k

点赞数 12

分类专栏： PTA 文章标签： c语言

本文链接：https://blog.csdn.net/weixin_73142957/article/details/129025057

版权

PTA 专栏收录该内容

41 篇文章 79 订阅

订阅专栏

前言：内容包括四大模块：题目，代码实现，大致思路，代码解读

题目：

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

输出样例2：

Error: 2 != 3

代码实现：

#include<stdio.h>
int main()
{
    int row1 = 0;
    int col1 = 0;
    scanf("%d %d", &row1, &col1);
    int arr1[100][100] = { 0 };
    int i = 0;
    //输入第一个矩阵
    for (i = 0; i < row1; i++)
    {
        int j = 0;
        for (j = 0; j < col1; j++)
        {
            scanf("%d", &arr1[i][j]);
        }
    }

    int row2 = 0;
    int col2 = 0;
    scanf("%d %d", &row2, &col2);
    int arr2[100][100] = { 0 };
    //输入第二个矩阵
    for (i = 0; i < row2; i++)
    {
        int j = 0;
        for (j = 0; j < col2; j++)
        {
            scanf("%d", &arr2[i][j]);
        }
    }
    //判断
    if (col1 != row2)
    {
        printf("Error: %d != %d", col1, row2);
    }
    else
    {
        //输出乘积矩阵
        printf("%d %d\n",row1,col2);
        for (i = 0; i < row1; i++)
        {
            int j = 0;
            for (j = 0; j < col2; j++)
            {
                int sum = 0;
                int k = 0;
                for (k = 0; k < row2; k++)
                {
                    sum += arr1[i][k] * arr2[k][j];
                }
                printf("%d", sum);
                if (j != col2 - 1)
                {
                    printf(" ");
                }
                else
                {
                    printf("\n");
                }
            }
        }
    }
    return 0;
}

大致思路：

1. 需要判断第一个矩阵arr1的列数col1和第二个矩阵arr2的行数row2是否相同

若不相同，则打印Error: Ca != Rb

若相同，则输出乘积矩阵AB

2. 输出乘积矩阵AB：

乘积矩阵AB的行是第一个矩阵arr1的行数row1，列是第二个矩阵的列col2

乘积矩阵AB第i行第j列的元素=第一个矩阵arr1第i行的每个数字对应乘以第二个矩阵arr2第j列的数字

如：arr1：1 2 3

4 5 6

arr2：7 8 9 0

-1 -2 -3 -4

5 6 7 8

乘积矩阵：20 22 24 16

53 58 63 28

第0行第0列的元素（20）：1*7+2*（-1）+3*5

第0行第1列的元素（22）：1*8+2*（-2）+3*6

第1行第0列的元素（53）：4*7+5*（-1）+6*5

……依次类推

代码解读：

part 1：输入第一个矩阵和第二个矩阵

    int row1 = 0;
    int col1 = 0;
    scanf("%d %d", &row1, &col1);
    int arr1[100][100] = { 0 };
    int i = 0;
    for (i = 0; i < row1; i++)
    {
        int j = 0;
        for (j = 0; j < col1; j++)
        {
            scanf("%d", &arr1[i][j]);
        }
    }

    int row2 = 0;
    int col2 = 0;
    scanf("%d %d", &row2, &col2);
    int arr2[100][100] = { 0 };
    for (i = 0; i < row2; i++)
    {
        int j = 0;
        for (j = 0; j < col2; j++)
        {
            scanf("%d", &arr2[i][j]);
        }
    }

part 2：输出乘积矩阵

    if (col1 != row2)
    {
        printf("Error: %d != %d", col1, row2);
    }
    else
    {
        printf("%d %d\n",row1,col2);
        for (i = 0; i < row1; i++)
        {
            int j = 0;
            for (j = 0; j < col2; j++)
            {
                int sum = 0;
                int k = 0;
                for (k = 0; k < row2; k++)
                {
                    sum += arr1[i][k] * arr2[k][j];
                }
                printf("%d", sum);
                if (j != col2 - 1)
                {
                    printf(" ");
                }
                else
                {
                    printf("\n");
                }
            }
        }
    }

1. 在输出乘积矩阵之前需要判断是否满足输出条件，即：Ca与Rb相等时，两个矩阵才能相乘

Ca是第一个矩阵的列数，Rb是第二个矩阵的行数

2. 满足条件，输出乘积矩阵

a. 构建乘积矩阵（二维数组）的大致框架：行数row1，列数col2

b. 按照思路中的算法求出乘积矩阵中每一个位置上的数字

c. 若打印出的数字不在乘积矩阵中最后一列的位置上，则在它后面输出空格

否则输出换行