题目:
你将获得 K 个鸡蛋,并可以使用一栋从 1 到 N 共有 N 层楼的建筑。
每个蛋的功能都是一样的,如果一个蛋碎了,你就不能再把它掉下去。
你知道存在楼层 F ,满足 0 <= F <= N 任何从高于 F 的楼层落下的鸡蛋都会碎,从 F 楼层或比它低的楼层落下的鸡蛋都不会破。
每次移动,你可以取一个鸡蛋(如果你有完整的鸡蛋)并把它从任一楼层 X 扔下(满足 1 <= X <= N)。
你的目标是确切地知道 F 的值是多少。
无论 F 的初始值如何,你确定 F 的值的最小移动次数是多少?
示例 1:
输入:K = 1, N = 2
输出:2
解释:
鸡蛋从 1 楼掉落。如果它碎了,我们肯定知道 F = 0 。
否则,鸡蛋从 2 楼掉落。如果它碎了,我们肯定知道 F = 1 。
如果它没碎,那么我们肯定知道 F = 2 。
因此,在最坏的情况下我们需要移动 2 次以确定 F 是多少。
示例 2:输入:K = 2, N = 6
输出:3
示例 3:输入:K = 3, N = 14
输出:4
思路:
动态规划+二分法
在 mid 层楼扔了鸡蛋之后,可能出现两种情况:鸡蛋碎了,鸡蛋没碎。
如果鸡蛋碎了,那么鸡蛋的个数
K
应该减一如果鸡蛋没碎,那么鸡蛋的个数
K
不变
代码实现:
/**
* @param {number} K
* @param {number} N
* @return {number}
*/
var superEggDrop = function (K, N) {
let memo = new Map();
var dp = (K, N) => {
if (K === 1) return N;
if (K === 0 || N === 0) return 0;
let value = memo.get(`${K}-${N}`);
if (value) {
return value;
}
let res = Infinity;
let left = 1, right = N;
while (left <= right) {
let mid = parseInt((left + right) / 2);
let broken = dp(K - 1, mid - 1); // 碎
let notBroken = dp(K, N - mid); // 没碎
if (broken > notBroken) {
right = mid - 1;
res = Math.min(res, broken + 1);
} else {
left = mid + 1;
res = Math.min(res, notBroken + 1);
}
}
memo.set(`${K}-${N}`, res);
return res;
}
return dp(K, N);
};