如何将列表分配为子列表，同时保持元素的原始顺序？

Question

如何将列表拆分为给定数量的列表，按顺序取出元素并将它们分配到子列表（因此不对列表进行分区）？

我想尽可能“好”地做到这一点（使用Java 8功能或Guava或类似的东西。

• 示例列表：

  [1 2 3 4 5 6 7]

• 应分为 3 份：

  [1 4 7]

  [2 5]

  [3 6]

• 应分为 2 ：

  [1 3 5 7]

  [2 4 6]

Answer 1

如果源列表支持高效的随机访问，就像

ArrayList

一样，你可以使用

IntStream.range(0, source.size()).boxed()
  .collect(groupingBy(i->i%listCount, LinkedHashMap::new, mapping(source::get, toList())));

例如

List<Integer> source=IntStream.range(0, 20).boxed().collect(toList());
System.out.println(source);
int listCount=5;

Map<Integer, List<Integer>> collect = IntStream.range(0, source.size()).boxed()
  .collect(groupingBy(i->i%listCount, LinkedHashMap::new, mapping(source::get, toList())));
// in case it really has to be a List:
List<List<Integer>> result=new ArrayList<>(collect.values());

result.forEach(System.out::println);

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
[0, 5, 10, 15]
[1, 6, 11, 16]
[2, 7, 12, 17]
[3, 8, 13, 18]
[4, 9, 14, 19]

Answer 2

类似这样的东西可以将所有列表放入地图中，然后您只需要将子列表从地图中取出

int count = 0;
Map<Integer, List<Integer>> mapLists = list.stream()
                            .peek(i -> count ++)
                            .collect(Collectors.groupingBy(i -> count % numOfSubLists))

使用番石榴的另一种方法

https://google.github.io/guava/releases/snapshot/api/docs/com/google/common/collect/Lists.html#partition(java.util.List,%20int)

List<List<Integer>> lists = Lists.partition(list, noOfPartitions);

Answer 3

这是即将推出的 Java 24 的解决方案，它根据 Stream Gatherers

 功能使用自定义 

Gatherer。 收集器将

Stream<Integer>

转换为

Stream<List<Integer>>

，其中每个列表都是所需的分组之一。

int n = 3;
List<Integer> list = List.of(1, 2, 3, 4, 5, 6, 7);

class State {
    int index;
    List<List<Integer>> subLists =
            Stream.<List<Integer>>generate(ArrayList::new).limit(n).toList();
}
List<List<Integer>> groups = list.stream()
        .gather(Gatherer.<Integer, State, List<Integer>>ofSequential(
                State::new,
                Gatherer.Integrator.ofGreedy(((state, element, downstream) -> {
                    state.subLists.get(state.index++ % n).add(element);
                    return true;
                })),
                (state, downstream) -> {
                    state.subLists.forEach((List<Integer> l) -> {
                        if (!l.isEmpty()) {
                            downstream.push(l);
                        }
                    });
                }))
        .toList();

System.out.println(groups);

此收集器的状态是下一个流元素的索引以及将元素分组到的列表的列表。

积分器根据索引模n选择适当的子列表，并将当前元素添加到其中。 积分器还负责每次递增索引。

整理器将所有子列表发送到结果流。 如果 n 大于列表的大小，则会出现空分组。 这些被丢弃而不是被发送到流中。