稀疏矩阵的乘法

/*
 * 思路：
 * 1. 使用Stream API进行矩阵操作。尽管Stream API对于处理矩阵乘法来说不如传统循环直观，但我们仍然可以尝试。
 * 2. 我们可以将矩阵转换为Stream，然后使用flatMap和map进行必要的乘法操作。
 */
import java.util.stream.IntStream;
 
public class SparseMatrixMultiplicationStream {
    public int[][] multiply(int[][] A, int[][] B) {
        int m = A.length;
        int n = A[0].length;
        int p = B[0].length;
        int[][] result = new int[m][p];
 
        IntStream.range(0, m).forEach(i -> {
            IntStream.range(0, n).filter(k -> A[i][k] != 0).forEach(k -> {
                IntStream.range(0, p).filter(j -> B[k][j] != 0).forEach(j -> {
                    result[i][j] += A[i][k] * B[k][j];
                });
            });
        });
 
        return result;
    }
}