原题链接
华为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);
}
}