集合分段实例

最近一个需求,一个集合,按每5个分组,小于5个也是1组

写了个小方法,记录下

1思路就是利用list的subList()方法,来分页
2使用stream也能分页

方法1 subList()方法

public static void main(String[] args) {
        List<List<Integer>> list = new ArrayList<>();
        //自定义数据
        ArrayList<Integer> listData = new ArrayList<>();
        listData.add(1);
        listData.add(2);
        listData.add(3);
        listData.add(4);
        listData.add(5);
        listData.add(6);
        listData.add(7);
        listData.add(8);
        listData.add(9);
        listData.add(10);
        listData.add(11);
        int count = ((listData.size() - 1) / 5) + 1;
        ArrayList<Integer> listData2;
        for (int i = 0; i < count; i++) {
            if (i == 0) {
                listData2 = new ArrayList<>(listData.subList(0, 5));
                list.add(listData2);
            } else if ((i + 1) * 5 > listData.size()) {
                listData2 = new ArrayList<>(listData.subList(i * 5, (i * 5) + (listData.size() % 5)));
                list.add(listData2);
            } else {
                listData2 = new ArrayList<>(listData.subList(i * 5, listData.size()-1));
                list.add(listData2);
            }
            System.out.println("第" + (i + 1) + "次:" + listData2);
        }
        System.out.print("总的:");
        System.out.println(list);
    }

抽成个方法

public static void main(String[] args){
//自定义数据
        ArrayList<Integer> listData = new ArrayList<>();
        listData.add(1);
        listData.add(2);
        listData.add(3);
        listData.add(4);
        listData.add(5);
        listData.add(6);
        listData.add(7);
        listData.add(8);
        listData.add(9);
        listData.add(10);
        listData.add(11);
List<List<Integer>>  list = subList(listData,5);
	System.out.print("总的:");
    System.out.println(list);
}

//方法
//参数 listData 集合
//参数 splitSize 分页的大小
public static List<List<Integer>> subList(ArrayList<Integer> listData, int splitSize) {
        List<List<Integer>> list = new ArrayList<>();
        int count = ((listData.size() - 1) / splitSize) + 1;
        ArrayList<Integer> listData2;
        for (int i = 0; i < count; i++) {
            if (i == 0) {
                listData2 = new ArrayList<>(listData.subList(0, splitSize));
                list.add(listData2);
            } else if ((i + 1) * splitSize > listData.size()) {
                listData2 = new ArrayList<>(listData.subList(i * splitSize, (i * splitSize) + (listData.size() % splitSize)));
                list.add(listData2);
            } else {
                listData2 = new ArrayList<>(listData.subList(i * splitSize, listData.size()-1));
                list.add(listData2);
            }
            System.out.println("第" + (i + 1) + "次:" + listData2);
        }
        return list;
    }

结果

第1次:[1, 2, 3, 4, 5]
第2次:[6, 7, 8, 9, 10]
第3次:[11]
总的:[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11]]

方法二 stream方式

public static void main(String[] args) {
        //自定义数据
        ArrayList<Integer> listData = new ArrayList<>();
        listData.add(1);
        listData.add(2);
        listData.add(3);
        listData.add(4);
        listData.add(5);
        listData.add(6);
        listData.add(7);
        listData.add(8);
        listData.add(9);
        listData.add(10);
        listData.add(11);

		//返回的数据,二维数组
		 int maxSize = listData.size()/5 + 1;
        List<Collection> splitList = splitList(listData,maxSize ,5);
        System.out.print("总的:");
        System.out.println(splitList);
    }


//stream方式
//参数 collection集合
//参数 maxSize 页数
//参数 splitSize 每页大小
 public static <T> List<Collection> splitList(Collection<T> collection, int maxSize, int splitSize) {
        if(CollectionUtils.isEmpty(collection)) {
            return Collections.emptyList();
        }
        return Stream.iterate(0, f -> f + 1)
                .limit(maxSize)
                .parallel()
                .map(a -> collection.parallelStream().skip(a * splitSize).limit(splitSize).collect(Collectors.toList()))
                .filter(b -> !b.isEmpty())
                .collect(Collectors.toList());
    }

结果

总的:[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11]]

注:

没什么难度,实现方式也有很多,但是每次写有点烦,还是记录下,
有别的方法大家可以分享下~