混合两个元素的 Java 8 Stream

Question

我在数组列表中有很多 Slot 类型的对象。

Slot类如下图-

Slot{
   int start;
   int end;
}

让类型列表List<Slot>被称为slots .插槽根据开始时间排序。一个时隙的结束时间可能等于下一个时隙的开始时间，但它们永远不会重叠。

有什么方法可以让我使用 Java 8 流遍历此列表，如果一个槽的结束时间与下一个槽的开始时间匹配，则合并两个槽并将它们输出到 ArrayList 中？

Answer 1

我的免费 StreamEx 完美支持这种场景增强标准 Stream API 的库。有一个 intervalMap能够将多个相邻流元素折叠为单个元素的中间操作。这是完整的示例:

// Slot class and sample data are taken from @Andreas answer
List<Slot> slots = Arrays.asList(new Slot(3, 5), new Slot(5, 7), 
                new Slot(8, 10), new Slot(10, 11), new Slot(11, 13));

List<Slot> result = StreamEx.of(slots)
        .intervalMap((s1, s2) -> s1.end == s2.start,
                     (s1, s2) -> new Slot(s1.start, s2.end))
        .toList();
System.out.println(result);
// Output: [3-7, 8-13]

intervalMap 方法有两个参数。第一个是 BiPredicate，它从输入流中接受两个相邻的元素，如果它们必须合并则返回 true(这里的条件是 s1.end == s2.start)。第二个参数是一个 BiFunction，它从合并的系列中获取第一个和最后一个元素并生成结果元素。

请注意，例如，如果您有 100 个应该合并为一个的相邻插槽，则此解决方案不会创建 100 个中间对象(就像@Misha 的回答一样，这仍然非常有趣)，它会跟踪第一个和最后一个插槽该系列立即忘记了中间的一次。当然这个解决方案是并行友好的。如果您有数千个输入槽，使用 .parallel() 可能会提高性能。

请注意，当前的实现将重新创建 Slot，即使它没有与任何东西合并。在这种情况下，BinaryOperator 两次接收相同的 Slot 参数。如果你想优化这种情况，你可以进行额外的检查，比如 s1 == s2 ? s1 : ...:

List<Slot> result = StreamEx.of(slots)
        .intervalMap((s1, s2) -> s1.end == s2.start,
                     (s1, s2) -> s1 == s2 ? s1 : new Slot(s1.start, s2.end))
        .toList();