华为OD机试题-分月饼（Java版-递归+回溯）

中秋节，公司要给员工分月饼。公司有𝑚个员工，买了n 个月饼，且m≤n。每个员工至少分到 1 个月饼，但可以分多个。
要求：
单人分到最多的月饼个数为Max1，单人分到第二多的月饼个数为Max2，需要满足Max1−Max2≤3。
单人分到倒数第二多的月饼个数为Max(n−1)，单人分到最少的月饼个数为Max(n)，需要满足Max(n−1)−Max(n)≤3。问有多少种分月饼的方法？
输入格式
第一行输入两个整数
m和𝑛表示:m个员工和n个月饼，m≤n。
输出格式
输出有多少种分月饼的方法。
样例输入
输入
2 4
输出
2

输入
3 5
输出
2

输入
3 12
输出
6

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int m = scanner.nextInt();
        int n = scanner.nextInt();
        scanner.close();
        
        System.out.println(divide(n, m));
    }

    public static int divide(int n, int m) {
        int[] arr = new int[m];
        for (int i = 0; i < m; i++) {
            arr[i] = 1;
        }
        if (n == m) {
            return 1;
        }
        return backtrack(m, n - m, 0, arr);
    }

    //通过递归与回溯，计算出在每个员工一个月饼的基础上、符合条件的的全排列所有可能
    private static int backtrack(int m, int remaining, int index, int[] arr) {
        // 所有员工都分配好了月饼时，看月饼是否分配完
        if (index == m) {
            if (remaining == 0) {
                return 1;
            } else {
                return 0;
            }
        }
        
        // 月饼提前分配完，看是否满足差额限制
        if (remaining == 0) {
            if (index == 0 || (index > 0 && arr[index] >= arr[index - 1] - 3)) {
                return 1;
            } else {
                return 0;
            }
        }
        
        int result = 0;
        // 尝试给当前位置分配从1到remaining的月饼
        for (int i = 1; i <= remaining; i++) {
            int value = arr[index] + i;
            // 检查当前分配是否满足条件
            if (index == 0 || (index > 0 && arr[index - 1] >= value && value >= arr[index - 1] - 3)) {
                arr[index] += i;
                // 递归分配剩余的月饼
                result += backtrack(m, remaining - i, index + 1, arr);
                // 回溯
                arr[index] -= i;
            }
        }
        return result;
    }
}