文章目录
Java 中 List 分片的 7种方法业务需求背景实现方法方法一:最基本的 for 循环实现方法二:利用 List 的 subList() 方法方法三:stream 流操作 filter 方法过滤方法四:使用 Google Guava 的 Lists.partition 方法方法五:使用 apache commons 的 ListUtils.partition 方法方法六:使用 Hutool 的 ListUtil.partition 方法方法七:使用 Java8 的 Collectors.partitioningBy() 方法
Java 中 List 分片的 7种方法
业务需求背景
最近的一个需求中需要将一个完整的 List 分割成两部分,比如:
集合 [1,2,3,4,5] 需要分割成 [1,2,3] 和 [4,5];集合 [1,2,3,4,5,6] 需要分割成 [1,2,3] 和 [4,5,6];实现方法
方法一:最基本的 for 循环实现
实现代码:
public static void divideList(List<String> list) { List<String> firstList = new ArrayList<>(); List<String> secondList = new ArrayList<>(); int length = list.size(); // 判断集合元素个数是奇数个 if (length % 2 == 1) { for (int i = 0; i < length; i++) { if (i < (length / 2 + 1)) { firstList.add(list.get(i)); } else { secondList.add(list.get(i)); } } } else { for (int i = 0; i < length; i++) { if (i < (length / 2)) { firstList.add(list.get(i)); } else { secondList.add(list.get(i)); } } } System.out.println("firstList"+firstList); System.out.println("secondList"+secondList); }
测试:
public static void main(String[] args) { List<String> list = new ArrayList<>(); list.add("1"); list.add("2"); list.add("3"); list.add("4"); list.add("5"); list.add("6"); divideList(list); System.out.println("========================"); List<String> list1 = new ArrayList<>(); list1.add("1"); list1.add("2"); list1.add("3"); list1.add("4"); list1.add("5"); divideList(list1); }
测试结果:
方法二:利用 List 的 subList() 方法
public static void divideList2(List<String> list) { int length = list.size(); int mid = length / 2; // 计算中间位置 List<String> firstList = new ArrayList<>(); List<String> secondList = new ArrayList<>(); // 判断集合元素个数是奇数个 if (length % 2 == 1) { firstList = list.subList(0, mid + 1); // 获取前半部分 list secondList = list.subList(mid + 1, list.size()); // 获取后半部分 list } else { firstList = list.subList(0, mid); // 获取前半部分 list secondList = list.subList(mid, list.size()); // 获取后半部分 list } System.out.println("firstList" + firstList); System.out.println("secondList" + secondList); }
测试:
public static void main(String[] args) { List<String> list = new ArrayList<>(); list.add("1"); list.add("2"); list.add("3"); list.add("4"); list.add("5"); list.add("6"); divideList2(list); System.out.println("========================"); List<String> list1 = new ArrayList<>(); list1.add("1"); list1.add("2"); list1.add("3"); list1.add("4"); list1.add("5"); divideList2(list1); }
测试结果:
方法三:stream 流操作 filter 方法过滤
public static void divideList3(List<String> list) { int length = list.size(); int mid = length / 2; List<String> firstList = new ArrayList<>(); List<String> secondList = new ArrayList<>(); // 判断集合元素个数是奇数个 if (length % 2 == 1) { firstList = list.stream().filter(str -> Integer.valueOf(str) <= mid + 1).collect(Collectors.toList()); // 获取前半部分 list secondList = list.stream().filter(str -> Integer.valueOf(str) > mid + 1).collect(Collectors.toList()); // 获取后半部分 list } else { firstList = list.stream().filter(str -> Integer.valueOf(str) <= mid).collect(Collectors.toList()); // 获取前半部分 list secondList = list.stream().filter(str -> Integer.valueOf(str) > mid).collect(Collectors.toList()); // 获取后半部分 list } System.out.println("firstList" + firstList); System.out.println("secondList" + secondList); }
测试:
public static void main(String[] args) { List<String> list = new ArrayList<>(); list.add("1"); list.add("2"); list.add("3"); list.add("4"); list.add("5"); list.add("6"); divideList3(list); System.out.println("========================"); List<String> list1 = new ArrayList<>(); list1.add("1"); list1.add("2"); list1.add("3"); list1.add("4"); list1.add("5"); divideList3(list1); }
测试结果:
方法四:使用 Google Guava 的 Lists.partition 方法
Lists.partition 方法可以更加灵活的按照几个元素为一组的形式进行分割。
首先,进入 Lists.partition 方法源码:
可以看到,方法需要传入两个参数,其中:list 参数是要分割的集合,size 表示几个元素分割为一组。比如 size 是3,表示拆分成拆分的集合大小为3(即3个元素分为一组),后面不足3的有多少算多少
。分割完集合,将分割完的集合封装在一个集合里返回。
在 pom.xml 中添加依赖:
<dependency> <groupId>com.google.guava</groupId> <artifactId>guava</artifactId> <version>31.0.1-jre</version> </dependency>
实现代码:
import com.google.common.collect.Lists;public static void divideList4(List<String> list) { int length = list.size(); int mid = length / 2; // size 是把集合拆分的大小,size 为表示拆分成拆分的集合大小为3, // 后面不足3的有多少算多少 List<List<String>> partition; if (length % 2 == 1) { partition = Lists.partition(list, mid + 1); } else { partition = Lists.partition(list, mid); } System.out.println(partition); }
测试:
public static void main(String[] args) { List<String> list = new ArrayList<>(); list.add("1"); list.add("2"); list.add("3"); list.add("4"); list.add("5"); list.add("6"); divideList4(list); System.out.println("========================"); List<String> list1 = new ArrayList<>(); list1.add("1"); list1.add("2"); list1.add("3"); list1.add("4"); list1.add("5"); divideList4(list1); }
测试结果:
方法五:使用 apache commons 的 ListUtils.partition 方法
apache commons 的 ListUtils.partition 方法使用和 Google Guava 的 Lists.partition 方法大体一致
添加依赖:
<dependency> <groupId>org.apache.commons</groupId> <artifactId>commons-collections4</artifactId> <version>4.4</version> </dependency>
实现代码:
import org.apache.commons.collections4.ListUtils;public static void divideList5(List<String> list) { int length = list.size(); int mid = length / 2; // size 是把集合拆分的大小,size 为表示拆分成拆分的集合大小为3, // 后面不足3的有多少算多少 List<List<String>> partition; if (length % 2 == 1) { partition = ListUtils.partition(list, mid + 1); } else { partition = ListUtils.partition(list, mid); } System.out.println(partition); }
测试:
public static void main(String[] args) { List<String> list = new ArrayList<>(); list.add("1"); list.add("2"); list.add("3"); list.add("4"); list.add("5"); list.add("6"); divideList5(list); System.out.println("========================"); List<String> list1 = new ArrayList<>(); list1.add("1"); list1.add("2"); list1.add("3"); list1.add("4"); list1.add("5"); divideList5(list1); }
测试结果:
方法六:使用 Hutool 的 ListUtil.partition 方法
添加依赖:
<dependency> <groupId>cn.hutool</groupId> <artifactId>hutool-all</artifactId> <version>5.7.14</version> </dependency>
实现代码:
import cn.hutool.core.collection.ListUtil;public static void divideList6(List<String> list) { int length = list.size(); int mid = length / 2; // size 是把集合拆分的大小,size 为表示拆分成拆分的集合大小为3, // 后面不足3的有多少算多少 List<List<String>> partition; if (length % 2 == 1) { partition = ListUtil.partition(list, mid + 1); } else { partition = ListUtil.partition(list, mid); } System.out.println(partition); }
方法七:使用 Java8 的 Collectors.partitioningBy() 方法
实现方法:
public static void divideList7(List<String> list) { Map<Boolean, List<String>> groups = list.stream().collect(Collectors.partitioningBy(s -> Integer.parseInt(s) > 3)); List<List<String>> subSets = new ArrayList<List<String>>(groups.values()); System.out.println(subSets); }
测试:
public static void main(String[] args) { List<String> list = new ArrayList<>(); list.add("1"); list.add("2"); list.add("3"); list.add("4"); list.add("5"); list.add("6"); divideList7(list); System.out.println("========================"); List<String> list1 = new ArrayList<>(); list1.add("1"); list1.add("2"); list1.add("3"); list1.add("4"); list1.add("5"); divideList7(list1); }
测试结果: