如何将嵌套循环转换为嵌套流？ [复制]

Question

假设我有一个 int 数组，我想找到总和为零的所有对。在过去，我会做这样的事情：

public static int count(int[] a) {
    int cnt = 0;
    int N = a.length;
    for (int i = 0; i < N; i++) {
        for (int j = i + 1; j < N; j++) {

        if (a[i] + a[j] == 0) {
            cnt++;
        }
       }
    }
    return cnt;
    }

现在我在将其转换为流时遇到了一些问题。

我尝试了以下方法，但得到了 IllegalStateException：

final IntStream stream1 = Arrays.stream(a);
final IntStream stream2 = Arrays.stream(a);
long res = stream1.flatMap(i -> stream2.filter(j -> (j + i == 0))).count();

Answer 1

将 stream2.filter 更改为 Arrays.stream(a) - 当您第二次尝试处理它时，Stream 将被关闭。

但无论如何你都没有正确计数，应该是：

 IntStream.range(0, a.length)
          .flatMap(x -> Arrays.stream(a).skip(x + 1).filter(j -> j + a[x] == 0))
          .count();

Answer 2

Streams 是不可重复使用的，因此您无法定义一次并继续重复使用。这就是为什么将它们存储在变量中被认为是一种不好的做法。

您需要在现场创建它们：

long res = Arrays.stream(a)
  .flatMap(i -> Arrays.stream(a).filter(j -> (j + i == 0)))
  .count();

另外，为了充分体现提供的实现并避免重复，您需要在每个嵌套的Stream 开头添加skip 一些元素：

long res = IntStream.range(0, a.length)
  .flatMap(i -> Arrays.stream(a).skip(i + 1).filter(j -> (j + a[i] == 0)))
  .count();

---

如果您想将它们提取到变量中，您需要使用Supplier，它将作为我们Streams 的工厂。每个get 调用都会创建一个新的Stream 实例：

final Supplier<IntStream> s = () -> Arrays.stream(a);
long res = s.get()
  .flatMap(i -> s.get().filter(j -> (j + i == 0)))
  .count();