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

时间:2016-06-08 12:36:46

标签: java java-8 java-stream

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

3 个答案:

答案 0 :(得分:9)

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

$response = $twitter->setPostfields(
    array('status' => $status)
)

这将获得给定列表中最大列表的大小。对于每个索引,然后将其与由该索引处的每个列表的元素形成的流进行平面映射(如果该元素存在)。

然后您可以将其与

一起使用
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());
}

使用protonpack库,您可以使用interleave方法执行此操作:

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]
}

答案 1 :(得分:0)

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

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

答案 2 :(得分:0)

我喜欢Tunaki的回答,并且已经对它做了更多的工作:

static <T> List<T> interleave(List<? extends Collection<T>> lists) {
    int maxSize = lists.stream().mapToInt(Collection::size).max().orElse(0);
    List<Iterator<T>> 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());
}

我想知道我们是否可以将初始列表视为流,但事实证明我们可以。作为副作用,我们在列表中避免使用get(i)(如果这真的很长LinkedList,我们会想要避免它)。如果不了解原始列表的最大长度,我就找不到办法。