碰巧解决PTA L1-048 矩阵A乘以B (15分) -- 我终于用Java解出来了

原创于 2020-10-29 20:37:43 发布 · 645 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #java

该代码实现读取两个矩阵并计算它们的乘积，如果规模不匹配则输出错误信息。程序首先读取每个矩阵的行数和列数，然后逐元素存储矩阵数据。在确保可以进行矩阵乘法之后，计算乘积并以指定格式输出结果。

L1-048 矩阵A乘以B (15分)

给定两个矩阵A和B，要求你计算它们的乘积矩阵AB。需要注意的是，只有规模匹配的矩阵才可以相乘。即若A有R
a
行、C
a
列，B有R
b
行、C
b
列，则只有C
a
与R
b
相等时，两个矩阵才能相乘。

输入格式：
输入先后给出两个矩阵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

我的解法：（我都傻了，卡的还是输入输出的点）

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        String[] s = reader.readLine().split(" ");
        int l1 = Integer.parseInt(s[0]);
        int c1 = Integer.parseInt(s[1]);
        int [][] nums1 = new int[l1][c1];
        for (int i = 0; i < l1; i++) {
            s = reader.readLine().split(" ");
            for (int j = 0; j < c1; j++) {
                nums1[i][j] = Integer.parseInt(s[j]);
            }
        }

        s = reader.readLine().split(" ");
        int l2 = Integer.parseInt(s[0]);
        int c2 = Integer.parseInt(s[1]);
        int [][] nums2 = new int[l2][c2];
        for (int i = 0; i < l2; i++) {
            s = reader.readLine().split(" ");
            for (int j = 0; j < c2; j++) {
                nums2[i][j] = Integer.parseInt(s[j]);
            }
        }

        if(c1 != l2) {
            System.out.printf("Error: %d != %d", c1, l2);
            return;
        }

        StringBuffer buffer = new StringBuffer();
        buffer.append(l1 + " " + c2 + "\n");
        for (int i = 0; i < l1; i++) {
            for (int j = 0; j < c2; j++) {
                int sum = 0;
                for (int k = 0; k < c1; k++) {
                    sum += nums1[i][k] * nums2[k][j];
                }
                buffer.append(sum);
                if(j == c2 - 1) continue;
                buffer.append(" ");
            }
            if(i == l1 - 1) continue;
            buffer.append("\n");
        }
        System.out.println(buffer.toString());
        reader.close();
    }

}