分月饼 java题解

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in); 
        int m = sc.nextInt(); // 读取员工数量m
        int n = sc.nextInt(); // 读取月饼数量n
        // 调用distribute方法并打印返回的分配方法总数
        //先默认每人分一个，修改remain和maxallow ，不计入后续计算。
        System.out.println(distribute(m, n-m, n-m, 0, new int[m]));
        sc.close();  
    }
    //m,distribution;remain,maxallow,index
    //员工数，分配计划；剩余月饼数，当前人可分配最大月饼数，当前人
    // 递归分配月饼的方法，从大到小分配，a>=b>=c...
    private static int distribute(int m, int remaining, int maxAllowed, int index, int[] distribution) {
        // 如果当前是最后一个员工，满足条件的话就完成了一次成功的分配
        if (index == m - 1) {
            // 检查剩余的月饼是否可以全部分配给最后一个员工
            // 并且满足最大和最小月饼数差值不超过3的条件
            if (remaining <= maxAllowed && (index == 0 || Math.abs(distribution[index - 1] - remaining) <= 3)) {
                return 1; // 如果满足条件，返回1种方法
            }
            return 0; // 如果不满足条件，返回0种方法
        }
        //对当前员工进行再分配
        int ways = 0; // 初始化分配方法数为0
        // 确定当前员工可以再分配的月饼数的起始值（每人本来就有1个）
        int start = (index == 0) ? 0 : distribution[index - 1] - 3;//上一个 - 当前 <= 3
        // 确定当前员工可以再分配的月饼数的结束值
        int end = Math.min(remaining, maxAllowed);//不能超过当前剩余总数，不能超过上限

        // 遍历所有可能的分配数量
        for (int i = start; i <= end; i++) {
            distribution[index] = i; // 尝试再分配i个月饼给当前员工
            // 如果当前分配满足最大和最小月饼数差值不超过3的条件
            if (index == 0 || Math.abs(distribution[index - 1] - i) <= 3) {
                // 递归调用distribute方法，计算剩余员工和月饼的分配方法
                ways += distribute(m, remaining - i, i, index + 1, distribution);
                //剩余月饼=上次剩余月饼-分给当前员工的；下一个员工的月饼不得大于当前员工，上限为当前员工的月饼数
            }
        }

        return ways; // 返回总的分配方法数
    }
}