public static void main(String[] args) {
List<User> originalList = new ArrayList<>();
int partitionSize = 3;
List<List<User>> partitionedList = IntStream.range(0, originalList.size()).boxed().
collect(Collectors.groupingBy(index -> index / partitionSize)).values().stream().
map(indices -> indices.stream().map(originalList::get).collect(Collectors.toList())).collect(Collectors.toList());
System.out.println(partitionedList);
}
首先将originalList均分成大小为3的子列表,并将它们存储在partitionedList中。
使用IntStream.range()方法生成原始列表的索引流,并使用Collectors.groupingBy()方法将索引分成分组。然后,它将每个分组中的索引映射为原始列表中的元素,并使用Collectors.toList()方法将它们收集到一个新的List中。最后,它将所有这些子列表收集到一个大列表中。
使用这种方法可以非常高效地均分列表,因为它只需要遍历原始列表一次,并且使用了Java 8的并行流特性,可以更快地处理大型列表。
其他方案:
public static <T> List<List<T>> splitList(List<T> list, int n) {
List<List<T>> result = new ArrayList<>();
int size = list.size();
int batchSize = (size + n - 1) / n;
int index = 0;
while (index < size) {
int nextIndex = Math.min(index + batchSize, size);
result.add(new ArrayList<>(list.subList(index, nextIndex)));
index = nextIndex;
}
return result;
}
-----开源项目----