java - int[][] 行的总和作为 int[] (Java)

Question

我试图获取二维数组中每行的总和并将这些值存储在一个新数组中。现在 sum[] 仅返回存储在第一行中的值。请帮助我理解我在这里缺少什么。

public static int[] rowSum(int[][] matrix)  //find sum of digits in given row


 {


  int[] sum = new int[6];

  for (int col = 0; col < ARRAY_LENGTH; col++)
  {

     for (int row = 0; row < ARRAY_LENGTH; row++)
     {

        sum[row] += matrix[col][row];
     }


  }

  return sum;


}

Answer 1

由于您使用 Java 8，您可以使用:

return Arrays.stream(matrix) // Stream<int[]>
    .mapToInt(row -> Arrays.stream(row).sum()) // IntStream
    .toArray(); // int[]