java - 如何使用 Java 流查找数组列总和

Question

我有一个二维整数数组:

int arr[][] = new int[rows][columns];

可以使用以下方法找到第 n 行总和:

int rSum=Arrays.stream(arr[n]).sum();

如何找到第 p 列总和？

int cSum=Arrays.stream(arr[][p]).sum();

上面一行行不通。

Answer 1

您可以将每一行映射到第 p 元素:

int cSum = Arrays.stream(arr).mapToInt(row -> row[p]).sum();

它的工作原理如下:首先我们从 arr 构造一个流。此流将包含“矩阵”的行。然后对于每个这样的 row，我们将其 mapToInt 为 int:该行的第 p 元素。然后我们将 sum(..) int 流加在一起求和。