static <A, B> Stream<Pair<A, B>> zip(Stream<A> as, Stream<B> bs)
{
Iterator<A> i=as.iterator();
return bs.filter(x->i.hasNext()).map(b->new Pair<>(i.next(), b));
}
这不提供并行执行,但原始 zip 实现也不提供。
作为F. Böller has pointed out,如果bs 是无限的而as 不是,则它不起作用。对于适用于无限和有限流的所有可能组合的解决方案,在hasNext 方法中检查两个源的中间Iterator 似乎是不可避免的¹:
static <A, B> Stream<Pair<A,B>> zip(Stream<A> as, Stream<B> bs) {
Iterator<A> i1 = as.iterator();
Iterator<B> i2 = bs.iterator();
Iterable<Pair<A,B>> i=()->new Iterator<Pair<A,B>>() {
public boolean hasNext() {
return i1.hasNext() && i2.hasNext();
}
public Pair<A,B> next() {
return new Pair<A,B>(i1.next(), i2.next());
}
};
return StreamSupport.stream(i.spliterator(), false);
}
如果您想要并行压缩,您应该考虑Stream 的来源。例如。你可以压缩两个ArrayLists(或任何RandomAccessList)喜欢
ArrayList<Foo> l1=new ArrayList<>();
ArrayList<Bar> l2=new ArrayList<>();
IntStream.range(0, Math.min(l1.size(), l2.size()))
.mapToObj(i->new Pair(l1.get(i), l2.get(i)))
. …
¹(除非您直接实现 Spliterator)