华为OD机试-分月饼

题目

公司分月饼，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+3和3+1是一种方案，则可以考虑递增发放，即一个人比一个人多

#include <iostream>

int m, n, res; // m个员工，n个月饼

// 四个参数分别代表，
// 第pos位员工
// 此时剩余的月饼数量
// 最多分到的数量和最少分到的数量
// 那么，第pos+1 个人获得的月饼数量的范围x1一定有
// x ≤ x1 ≤ min(sum − x, x + 3) 如果pos==m，且sum=0，则找到了，res+=1
void dfs(int pos, int sum, int minimum, int maximum)
{
	if (pos == m)
	{
		if (sum == 0) {
			res++;
		}
		return;
	}
	//maximum < minimum 保证递增
	if (minimum > sum || sum < 0 || maximum < minimum)
	{
		return; // 不合法的情况
	}

	for (int i = minimum; i <= maximum; i++) // 枚举下一个人获得的月饼数量
	{
		dfs(pos + 1, sum - i, i, std::min(sum - i, i + 3));
	}
}

int main()
{
	std::cin >> m >> n;
	dfs(0, n, 1, n);
	std::cout << res << std::endl;
	return 0;
}