分组值序列

Question

我想知道是否有任何巧妙的方法可以使用新的Stream API将值序列“分组”。

例如，将一系列整数分成整数组，其中每个组都是一个升序的数字序列：

IntStream seq = IntStream.of(1, 2, 3, -1, -1, 1, 2, 1, 2);
IntFunction next = i -> i + 1;

// DESIRED OUTPUT: [[1,2,3], [-1], [-1], [1,2], [1,2]]

Answer 1

不幸的是，Stream API并不是非常适合解决涉及Stream元素的依赖操作的问题，例如此元素。

但是，您可以为此使用StreamEx库：

public static void main(String[] args) {
    IntStream seq = IntStream.of(1, 2, 3, -1, -1, 1, 2, 1, 2);
    IntUnaryOperator next = i -> i + 1;

    List<List<Integer>> result = 
        IntStreamEx.of(seq).boxed().groupRuns((i1, i2) -> next.applyAsInt(i1) == i2).toList();

    System.out.println(result); // prints "[[1, 2, 3], [-1], [-1], [1, 2], [1, 2]]"
}

这将分组为List所有连续的整数，其中第二个等于next应用于第一个的函数。最后，将此流收集到中List。