牛客-NC143-矩阵乘法

方法一：模拟法

思路：可以直接模拟得到答案，需要注意的是：（1）a数组的列数应该和b数组的行数相同才能进行矩阵相乘；（2）新产生的二维数组应该是a数组的行与b数组的列对应的元素相乘得到的。于是，可以写出以下代码：

import java.util.*;


public class Solution {
   
    /** * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可 * * @param a int整型二维数组 第一个矩阵 * @param b int整型二维数组 第二个矩阵 * @return int整型二维数组 */
    public int[][] solve (int[][] a, int[][] b) {
   
        // write code here
        int rowA = a.length; // a的行数
        int colB = b[0].length; // b的列数
        int p = a[0].length; // a的列数/b的行数
        int[][] ret = new int[rowA][colB];
        for (int i = 0; i < rowA; i++) {
   
            for (int j = 0; j < colB; j++) {
   
                // 计算新列表存放的值
                for (int k = 0; k < p; k++) {
   
                    ret[i][j] += a[i][k] * b[k][j];
                }
            }
        }
        return ret;
    }
}

时间复杂度： O(max(M + N) * P)，M，N分别是a和b的行数和列数，P是a的列数或者b的行数。
空间复杂度： O(1)，常量级空间复杂度（因为我们必须返回计算结果，所以用来存结果的ret并不用算额外的空间）。