华为OD机考题(HJ69 矩阵乘法)

前言

经过前期的数据结构和算法学习，开始以OD机考题作为练习题，继续加强下熟练程度。

描述

如果A是个x行y列的矩阵，B是个y行z列的矩阵，把A和B相乘，其结果将是另一个x行z列的矩阵C。这个矩阵的每个元素是由下面的公式决定的

矩阵的大小不超过100*100

输入描述：

第一行包含一个正整数x，代表第一个矩阵的行数
第二行包含一个正整数y，代表第一个矩阵的列数和第二个矩阵的行数
第三行包含一个正整数z，代表第二个矩阵的列数
之后x行，每行y个整数，代表第一个矩阵的值
之后y行，每行z个整数，代表第二个矩阵的值
 

输出描述：

对于每组输入数据，输出x行，每行z个整数，代表两个矩阵相乘的结果

示例1

输入：

2
3
2
1 2 3
3 2 1
1 2
2 1
3 3

输出：

14 13
10 11

说明：

1 2 3
3 2 1 
乘以
1 2
2 1
3 3
等于
14 13
10 11    

实现原理与步骤

1.两个矩阵的构建

2.3层循环执行矩阵的相乘，第一层为第一个矩阵的行下标，第二层为第二个矩阵的列下标，第三层则为第一个矩阵的列(同时也是第二个矩阵的行)。

3.打印结果矩阵

实现代码

import java.util.Scanner;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 注意 hasNext 和 hasNextLine 的区别
        int rowA = in.nextInt();
        int colA = in.nextInt();
        int rowB=colA;
        int colB= in.nextInt();
        int[][] oneGrid = new int[rowA][colA];
        for (int i = 0; i < rowA; i++) {
            for (int j = 0; j < colA; j++) {
                oneGrid[i][j] = in.nextInt();
            }
        }
        int[][] twoGrid = new int[rowB][colB];
        for (int i = 0; i < rowB; i++) {
            for (int j = 0; j < colB; j++) {
                twoGrid[i][j] = in.nextInt();
            }
        }

        int[][] res = new int[rowA][colB];

        for (int i = 0; i < rowA; i++) {
            for (int j = 0; j < colB; j++) {
                for (int k = 0; k < colA; k++) {
                    res[i][j] += oneGrid[i][k] * twoGrid[k][j];
                }
            }
        }

        for (int i = 0; i < res.length; i++) {
            for (int j = 0; j < res[0].length; j++) {
                System.out.print(res[i][j]+" ");
            }
            System.out.println();
        }

    }
}

1.QA: