是否可以将.flatMap()或.collect()均匀分布在多个集合中

例如,有[1,2,3,4,5][6,7,8][9,0] 。 有没有办法避免循环使用迭代器通过Java 8流API交错这些集合以获得以下结果 – [1,6,9,2,7,0,3,8,4,5]

我不确定Stream API是否有更简单的方法,但您可以使用所有列表的索引上的流来考虑:

 static  List interleave(List> 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 list1 = Arrays.asList(1,2,3,4,5); List list2 = Arrays.asList(6,7,8); List 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> lists = Arrays.asList(list1.stream(), list2.stream(), list3.stream()); List result = StreamUtils.interleave(Selectors.roundRobin(), lists).collect(Collectors.toList()); System.out.println(result); 

虽然不是一个很好的解决方案。 试一试

  List listA = Arrays.asList("a1","a2","a3","a4"); List listB = Arrays.asList("b1","b2","b3","b4"); List listC = Arrays.asList("c1","c2","c3","c4"); List combined = Stream.of(listA, listB, listC).flatMap(Collection::stream).collect(Collectors.toList()); AtomicInteger index = new AtomicInteger(); List list1 = combined .stream().filter(x -> index.incrementAndGet() % 4 ==1).collect(Collectors.toList()); index.set(0); List list2 = combined .stream().filter(x -> index.incrementAndGet() % 4 ==2).collect(Collectors.toList()); index.set(0); List list3 = combined .stream().filter(x -> index.incrementAndGet() % 4 ==3).collect(Collectors.toList()); index.set(0); List list0 = combined .stream().filter(x -> index.incrementAndGet() % 4 ==0).collect(Collectors.toList()); List desiredOutput = Stream.of(list1, list2, list3,list0).flatMap(Collection::stream).collect(Collectors.toList()); System.out.println(String.join(",", desiredOutput)); 

我喜欢Tunaki的回答,并且已经开展了更多工作:

 static  List interleave(List> lists) { int maxSize = lists.stream().mapToInt(Collection::size).max().orElse(0); List> iterators = lists.stream().map(l -> l.stream().iterator()).collect(Collectors.toList()); return IntStream.range(0, maxSize) .boxed() .flatMap(i -> iterators.stream().filter(it -> it.hasNext()).map(it -> it.next())) .collect(Collectors.toList()); } 

我想看看我们是否可以将初始列表视为流,但事实certificate我们可以。 作为一个副作用,我们避免get(i)在列表上(如果那是一个非常长的LinkedList ,我们想要避免它)。 如果不了解原始列表的最大长度,我就找不到办法。