.flatMap（）或.collect（）是否可以平均多个集合

Question

比如有收藏[1,2,3,4,5]，[6,7,8]，[9,0]。任何避免迭代器循环的方法都可以通过Java 8流API交错这些集合以获得以下结果-
[1,6,9,2,7,0,3,8,4,5]吗？

Answer 1

我不确定Stream API是否有更简单的方法，但是您可以在所有列表的索引上使用流来考虑以下问题：

static <T> List<T> interleave(List<List<T>> lists) {
    int maxSize = lists.stream().mapToInt(List::size).max().orElse(0);
    return IntStream.range(0, maxSize)
                    .boxed()
                    .flatMap(i -> lists.stream().filter(l -> i < l.size()).map(l -> l.get(i)))
                    .collect(Collectors.toList());
}

这将获得给定列表中最大列表的大小。然后，对于每个索引，它使用该列表中每个列表的元素在该索引处形成的流（如果该元素存在的话）进行平面映射。

然后，您可以将其与

public static void main(String[] args) {
    List<Integer> list1 = Arrays.asList(1,2,3,4,5);
    List<Integer> list2 = Arrays.asList(6,7,8);
    List<Integer> list3 = Arrays.asList(9,0);
    System.out.println(interleave(Arrays.asList(list1, list2, list3))); // [1, 6, 9, 2, 7, 0, 3, 8, 4, 5]
}

使用protonpack库，您可以使用方法interleave来做到这一点：

List<Stream<Integer>> lists = Arrays.asList(list1.stream(), list2.stream(), list3.stream());
List<Integer> result = StreamUtils.interleave(Selectors.roundRobin(), lists).collect(Collectors.toList());
System.out.println(result);