华为OD机试真题-分月饼-java代码实现(递归回溯实现)

原题链接

华为OD机试真题-分月饼-2023年OD统一考试（C卷）_分月饼华为-CSDN博客

思路

1.尝试将10块月饼分给3名员工.为了方便,给员工编号1,2,3

2.由少到多分配月饼,后分配的员工月饼数量不少于前一个分配的员工,并且不能多3块.

3.第一次尝试,为了满足题意,最少可以给1号员工1块月饼,那么2号员工可以分的{1,2,3,4}块月饼

对应的3号员工可以分的{1,2,3,4},{2,3,4,5},{3,4,5,6},{4,5,6,7}块月饼.但是月饼的总数有限,所以

最后分月饼的员工月饼数量不能大于剩余的月饼数,所以3号员工应该分得{1,2,3,4},{2,3,4,5},{3,4,5,6},{4,5,5,5}块月饼

4.最多可以给第一号员工 10 / 3 = 3 块数量的月饼.如果分配4块,至少需要12块月饼,但月饼总数是10,所以10 / 3向下取整.最多给3块.

5.在第i次尝试中,第i号员工能分的的月饼数量,应该由i - 1员工分到的数量last与当前剩余月饼数决定,最少分得与i - 1个员工相同的月饼,最多分的Math.min(last+3,剩余数量)块月饼

java代码

递归入口

   /**
     * @param n n 块月饼 n > m
     * @param m 员工人数
     */
    private static void solution(int n, int m) {
        //当前分支
        ArrayList<Integer> current = new ArrayList<>();
        //保存结果
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        //递归,结果存在result里
        process(n, m, 0, current, result);
        System.out.println(result.size());
    }

递归代码

/**
     * @param n       剩余月饼数
     * @param m       员工人数
     * @param last    给了前一个员工多少块月饼
     * @param current 当前分支下月饼分配情况
     * @param result  结果
     */
    private static void process(int n, int m, int last, ArrayList<Integer> current, ArrayList<ArrayList<Integer>> result) {

        //退出条件
        if (m == current.size()) {
            //月饼分完,是我们要的结果
            if (n == 0) {
                result.add(new ArrayList<>(current));
            }
            //没有分完,或者分多了,不存入结果
            return;
        }
        //如果第一次分月饼,可以从1开始分配
        //如果不是第一次分配月饼,最少不能低于当前分支的上一个分配的数量
        int min = current.isEmpty() ? 1 : Math.max(last, 1);
        //如果是第一次分月饼,为了满足每人分一块,最多可以给n / m 块月饼
        //如果不是第一次分月饼,最多可以根据前一次分配的月饼数last 加上3 ,但不能超过剩余月饼数
        //last 表示当前分支前一个节点分配了多少个月饼,等于下面for循环的i,也等于current.get(current.size() - 1),
        int max = current.isEmpty() ? n / m : Math.min(n, last + 3);

        for (int i = min; i <= max; i++) {
            //当前分支下,这个节点上的员工给i个月饼
            current.add(i);
            process(n - i, m, i, current, result);
            //回溯
            current.remove(current.size() - 1);
        }
    }