Divide a list into n lists

/**
     * Divide a list into n lists, mainly implemented by offset
     * @param source
     * @return
     */
    public static <T> List<List<T>> averageAssign(List<T> source, int n){
        List<List<T>> result = new ArrayList<List<T>>();
        int remaider = source.size() % n; //(Calculate the remainder first)
        int number = source.size( ) / n; // then quotient
        int offset = 0; // offset
        for (int i = 0; i < n; i++) {
            List<T> value = null;
            if (remaider > 0) {
                value = source.subList(i * number + offset, (i + 1) * number + offset + 1);
                remaider--;
                offset++;
            } else {
                value = source.subList(i * number + offset, (i + 1) * number + offset);
            }
            result.add(value);
        }
        return result;
    }

 

    @SuppressWarnings("unused")
    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);
        List<List<Integer>> lists = averageAssign(integers, 2);
        System.out.println(lists);
    }

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=325587266&siteId=291194637