如何交错(合并)两个Java 8流?
问题内容:
Stream<String> a = Stream.of("one", "three", "five");
Stream<String> b = Stream.of("two", "four", "six");
对于下面的输出,我需要做什么?
// one
// two
// three
// four
// five
// six
我调查了一下,concat
但是正如javadoc所解释的,它只是一个接一个地追加,它不会交错/散布。
Stream<String> out = Stream.concat(a, b);
out.forEach(System.out::println);
创建一个延迟串联的流,其元素是第一个流的所有元素,后跟第二个流的所有元素。
错误地给
// one
// three
// five
// two
// four
// six
如果我收集它们并进行迭代,但是希望获得更多Java8-y,Streamy :-),可以这样做
注意
我不想压缩流
“ zip”操作将从每个集合中获取一个元素并将其组合。
压缩操作的结果将是这样的:(不需要)
// onetwo
// threefour
// fivesix
问题答案:
我会用这样的东西:
public static <T> Stream<T> interleave(Stream<? extends T> a, Stream<? extends T> b) {
Spliterator<? extends T> spA = a.spliterator(), spB = b.spliterator();
long s = spA.estimateSize() + spB.estimateSize();
if(s < 0) s = Long.MAX_VALUE;
int ch = spA.characteristics() & spB.characteristics()
& (Spliterator.NONNULL|Spliterator.SIZED);
ch |= Spliterator.ORDERED;
return StreamSupport.stream(new Spliterators.AbstractSpliterator<T>(s, ch) {
Spliterator<? extends T> sp1 = spA, sp2 = spB;
@Override
public boolean tryAdvance(Consumer<? super T> action) {
Spliterator<? extends T> sp = sp1;
if(sp.tryAdvance(action)) {
sp1 = sp2;
sp2 = sp;
return true;
}
return sp2.tryAdvance(action);
}
}, false);
}
它尽可能保留输入流的特征,从而允许进行某些优化(例如count()
和toArray()
)。此外,ORDERED
当输入流可能无序时,它还会添加偶数,以反映交织。
当一个流中的元素多于另一个时,其余元素将出现在末尾。