Java8使用Lambda将一个List均分成指定n个List

public class Test {
    
    
    /**
     * 计算切分次数
     */
    private static Integer countStep(Integer size, int input) {
    
    

        return (size + input - 1) / input;
    }

    /** 
     * @param list  需要分隔的 集合
     * @param input 指定分隔size
     * @return
     */
    public static List<List<Integer>> split(List<Integer> list, int input) {
    
    
        int limit = countStep(list.size(), input);
        List<List<Integer>> splitList;
        splitList = Stream.iterate(0, n -> n + 1).limit(limit).
                 map(a -> list.stream().skip(a * input).limit(input).collect(Collectors.toList())).
                collect(Collectors.toList());
        //当输入数量小于分隔数量需反转
        if (input<limit){
    
    
            splitList = Stream.iterate(0, n -> n + 1).limit(input).
                    map(a -> list.stream().skip(a * limit).limit(limit).collect(Collectors.toList())).
                    collect(Collectors.toList());
        }


        return splitList;
    }

    public static void main(String[] args) {
    
    
        List<Integer> integers = new ArrayList<>();
        integers.add(1);
        integers.add(2);
        integers.add(3);
        integers.add(4);
        integers.add(5);
        integers.add(6);
        integers.add(7);
        integers.add(4);


        List<List<Integer>> split = split(integers, 2);
        System.out.println(split);
    }
}

输出

[[1, 2, 3, 4], [5, 6, 7, 4]]

猜你喜欢

转载自blog.csdn.net/qq_42261668/article/details/108528900