📜  矩阵乘法 - Java (1)

📅  最后修改于: 2023-12-03 14:56:29.729000             🧑  作者: Mango

矩阵乘法 - Java

矩阵乘法是计算机科学中常见的操作之一。它是在两个矩阵之间进行的,其中一个矩阵是具有m行n列,另一个矩阵是具有n行k列的。矩阵乘法结果是一个具有m行k列的矩阵。

在这个指南中,我们将学习如何在Java中实现矩阵乘法。

矩阵乘法的实现

假设我们有两个矩阵A和B,它们的尺寸分别是m x n和n x k。我们将创建一个结果矩阵C,其尺寸为m x k。对于C中的每个元素 [i][j],我们需要计算以下公式:

matrix_multiplication_formula

以下是Java代码,用于执行矩阵乘法:

public static int[][] matrixMultiplication(int[][] A, int[][] B) {
  int m = A.length;
  int n = A[0].length;
  int k = B[0].length;
  int[][] C = new int[m][k];
  
  for (int i = 0; i < m; i++) {
    for (int j = 0; j < k; j++) {
      for (int l = 0; l < n; l++) {
        C[i][j] += A[i][l] * B[l][j];
      }
    }
  }
  
  return C;
}
示例

以下是如何使用矩阵乘法方法的示例代码:

int[][] A = {{1, 2}, {3, 4}};
int[][] B = {{5, 6}, {7, 8}};

int[][] C = matrixMultiplication(A, B);

// 输出结果矩阵
for (int i = 0; i < C.length; i++) {
  for (int j = 0; j < C[0].length; j++) {
    System.out.print(C[i][j] + " ");
  }
  System.out.println();
}

以上代码将输出以下矩阵:

19 22 
43 50 
总结

矩阵乘法是计算机科学中一项重要的操作,能够在许多领域中使用。在Java中实现矩阵乘法需要进行多重循环,因此可以使用并行化优化来提高性能。