java 8 带流的嵌套循环

Question

我有一个 for 循环迭代 Integer [][]map 。 目前是这样的:

for(int i = 0; i < rows; i++) {
    for(int j = 0; j < columns; j++) {
        if(map[i][j] == 1)
            q.add(new Point(i,j));
    }
}        

假设我有 List<List<Integer>> maps2d 而不是二维数组。 我该如何使用流来做到这一点？

到目前为止我得到了这个:

maps2d.stream()
      .forEach(maps1d -> maps1d.stream()
                               .filter(u -> u == 1)
                               .forEach(u -> {

                               }
      )
);

到目前为止正确吗？ 如果是，我怎么算i和j为了创建new Point(i,j)并将其添加到 q ？

Answer 1

如果您确实想将流用于相同目的，那么一种选择是使用嵌套的 IntStream 来迭代索引。举个例子:

public static List<Point> foo(List<List<Integer>> map) {
  return IntStream.range(0, map.size()) // IntStream
      .mapToObj(
          i ->
              IntStream.range(0, map.get(i).size())
                  .filter(j -> map.get(i).get(j) == 1)
                  .mapToObj(j -> new Point(i, j))) // Stream<Stream<Point>>
      .flatMap(Function.identity()) // Stream<Point>
      .collect(Collectors.toList()); // List<Point>
}

就我个人而言，我不认为它具有很强的可读性。请注意，您仍然可以在列表中使用嵌套 for 循环，类似于您当前的解决方案:

public static List<Point> foo(List<List<Integer>> map) {
  List<Point> result = new ArrayList<>();
  for (int i = 0; i < map.size(); i++) {
    List<Integer> inner = map.get(i);
    for (int j = 0; j < inner.size(); j++) {
      if (inner.get(j) == 1) {
        result.add(new Point(i, j));
      }
    }
  }
  return result;
}