华为od（D卷）整数划分排序/员工分月饼 （转载）

转载原文链接：https://blog.csdn.net/qq_45763343/article/details/135894692
感谢原文博主。

题目描述

公司分月饼，m个员工，买了n个月饼，m <= n，每个员工至少分一个月饼，但是也可以分到多个，单人分到最多月饼的个数是Max1，单人分到第二多月饼个数是Max2。

但需要满足Max1-Max2 <= 3，单人分到第n-1多月饼个数是Max(n-1)，单人分到第n多月饼个数是Max(n), 想要满足Max(n-1) - Max(n) <= 3，问有多少种分月饼的方法？

输入描述

每一行输入m，n，表示m个员工，n个月饼，m <=n

输出描述

输出有多少种分法

示例1

输入
2 4

输出
2

说明
4=1+3
4=2+2
注意：1+3和3+1要算成同一种分法

示例2

输入
3 5

输出
2

说明
5=1+1+3
5=1+2+3

示例3

输入
3 12

输出
6

说明
满足要求的6种分法：
1、12 = 1 + 1 + 10 （Max1=10， Max2=1，不满足Max1-Max2 <= 3的约束）
2、12 = 1 + 2 + 9 （Max1=9，Max2=2，不满足Max1-Max2 <= 3的约束）
3、12 = 1 + 3 + 8 （Max1=8，Max2=3，不满足Max1-Max2 <= 3的约束）
4、12 = 1 + 4 + 7 （Max1=7，Max2=4，Max3=1， 满足要求）
5、12 = 1 + 5 + 6 （Max1=6，Max2=5，Max3=1， 不满足要求）
6、12 = 2 + 2 + 8 （Max1=8，Max2=2，不满足要求）
7、12 = 2 + 3 + 7 （Max1=7，Max2=3，不满足要求）
8、12 = 2 + 4 + 6 （Max1=6，Max2=4，Max3=2， 满足要求）
9、12 = 2 + 5 + 5 （Max1=5，Max2=2 满足要求）
10、12 = 3 + 3 + 6 （Max1=6，Max2=3 满足要求）
11、12 = 3 + 4 + 5 （Max1=5，Max2=4，Max3=3 满足要求）
12 = 4 + 4 + 4 （Max1=4，满足要求）

思路

转载博主已经给了很好的解析。

1. dp数组含义： dp[i][j][k] 表示前 i 个人分配了 j 个月饼，且第 i（最后一个人） 个人分配了 k 个月饼的方案数。

dp数组考虑所有因素，所以人和月饼数量是必须有， 而最后一个人的月饼数量，影响后人分配，所以也需要考虑。

3. 递推公式 dp[i][j][k] = dp[i-1][j-k][k-1]+dp[i-1][j-k][k-2]+dp[i-1][j-k][k-3]

第i个人分配的月饼数量和前一个人有关系，可能有三种情况得来：

1. 前一个人得到了k-1个的方案数
2. 前一个人得到了k-2个的方案数
3. 前一个人得到了k-3个的方案数

4. 初始化 dp[1][j][k] = 1;

一个人只有一种方案

代码

public class Demo10 {

    public static int countWays(int m, int n) {
        int[][][] dp = new int[m + 1][n + 1][n + 1];
        // 初始化
        for (int k = 1; k <= n; k++) {
            dp[1][k][k] = 1; // 只有1个员工时，只有一种分法
        }
        // 动态规划
        for (int i = 1; i <= m; i++) { // 人
            for (int j = i; j <= n; j++) { // 月饼数量   不少于员工数
                for (int k = 1; k <= j; k++) { // 每个员工至少分得1个月饼
                    for (int l = Math.max(1, k - 3); l <= k; l++) {
                        dp[i][j][k] += dp[i - 1][j - k][l];
                    }
                }
            }
        }

        // 汇总结果
        int count = 0;
        for (int k = 0; k <= n; k++) {
            count += dp[m][n][k];
        }

        return count;
    }

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int people = in.nextInt();
        int yuebing = in.nextInt();
        System.out.println(countWays(people, yuebing));
        in.close();
    }
}