迹忆客 专注技术分享

在本文中，我们将学习在 Java 中将两个矩阵相乘。

Java 中两个矩阵相乘

我们使用乘法和加法运算符来乘两个矩阵。 用户应按照以下步骤来将两个矩阵相乘。

• 创建两个矩阵并用值初始化它们。
• 创建一个矩阵来存储输出并用零对其进行初始化。
• 使用嵌套循环将第一个矩阵的每一行与第二个矩阵的每一列相乘。 将其总和存储在输出矩阵中。

示例代码：

class Test{
    public static void main(String[] args) {
    // create matrices to multiply
    int[][] a = {{11,2,3},{1,2,13},{1,12,3}};
    int[][] b ={{14,5,6},{4,5,6},{4,5,6}};
    // initialize array with zeros to store the result
    int[][] c = {{0,0,0},{0,0,0},{0,0,0}};
    // multiply A and B matrix
    for (int i = 0; i < a.length; ++i) {
        for (int j = 0; j < b.length; ++j) {
            // Initialize with 0, if not already
            c[i][j] = 0;
            // Multiply the ith row of A matrix and the jth column of B matrix and store its sum to C matrix as a result
            for (int k = 0; k < a.length; ++k) {
                c[i][j] += a[i][k] * b[k][j];
            }
        }
    }
    // Print the resultant matrix
    System.out.println("Matrix Multiplication of A and B is  ");
    for (int i = 0; i < a.length; ++i) {
        for (int j = 0; j < b.length; ++j) {
            System.out.print(c[i][j]);
            System.out.print("\t");
        }
        System.out.println("\n");
    }
    }
}

输出:

Matrix Multiplication of A and B is
174    80    96

74    80    96

74    80    96

矩阵乘法的时间复杂度

矩阵乘法的时间复杂度为 O(M^3)，因为我们使用 3 个嵌套循环。

矩阵乘法的空间复杂度

矩阵乘法算法的空间复杂度为 O(M^2)。