题目:
你将获得 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
提示:
1 <= K <= 100
1 <= N <= 10000
思路:
采用动态规划的方式
问题抽象:蛋多蛋少,层多层少的问题
base case:
1.当鸡蛋只有一个时,N=100,需要从第一层一层一层找。最差情况扔100(N)次。
2.无限多个蛋时,N=100
采用二分法
2^M=100 M>=6.64 最差扔7次
3.两个蛋,两层楼
F=10 碎了 0~10 不碎
F=20
以此类推
A蛋:10,20,30.....100
B蛋:x1,x2,x3.....x9 最差19次
4.令扔的间隔不等
n,n-1,n-2,....1
1+2+3....+n>=100 n=14
A蛋:14,27,39,50,60,69,77,84,90,95,99,100
B蛋:检验间隔
次数12~14次
4.K个蛋,N层楼 M次 M(K,N)=?
k\n | 1 | 2 | 3 |
---|---|---|---|
1 | 1 | ||
2 | 1 | ||
3 | 1 | ||
4 | 1 |
思路如图
暴力解决方式
找到状态转移方程
res = Math.min(res, Math.max(
dp(K - 1, i - 1),
dp(K, N - i)
) + 1);
class Solution {
public int superEggDrop(int K, int N) {
return dp(K,N);
}
private int dp(int K, int N) {
int[][] memo = new int[K+1][N+1];
//base case
if (K == 1) return N;
if (N == 0) return 0;
int res = Integer.MAX_VALUE;
//穷举所有可能出现的选择
for (int i = 1; i <= N; i++) {
res = Math.min(res, Math.max(
dp(K - 1, i - 1),
dp(K, N - i)
) + 1);
}
return res;
}
}
加个备忘录
class Solution {
public int superEggDrop(int K, int N) {
return dp(K, N);
}
private int dp(int K, int N) {
int memo[][] = new int[K + 1][N + 1];
/* if(K==1|N==0|N==1){
return N;
}*/
for (int i = 1; i <= N; i++) {
//only one egg
memo[1][i] = i;
//no egg
memo[0][i] = 0;
}
for (int i = 1; i <= K; i++) {
//only zero floor
memo[i][0] = 0;
}
for (int k = 2; k <= K; k++) {
//start from second egg
for (int n = 1; n <= N; n++) {
int min = N * N;
for (int x = 1; x <= n; x++) {
min = Math.min(min, Math.max(dp(K - 1, x - 1), dp(K, N - x)) + 1);
}
memo[K][N] = min;
}
}
return memo[K][N];
}
}
采用二分法优化速度
class Solution {
Map<Integer, Integer> cache = new HashMap<>();
public int superEggDrop(int K, int N) {
if (N == 0)
return 0;
else if (K == 1)
return N;
Integer key = N * 1000 + K; // K <= 100
if (cache.containsKey(key))
return cache.get(key);
int low = 1, high = N;
while (low + 1 < high) {
int middle = (low + high) / 2;
int lowVal = superEggDrop(K - 1, middle - 1);
int highVal = superEggDrop(K, N - middle);
if (lowVal < highVal)
low = middle;
else if (lowVal > highVal)
high = middle;
else
low = high = middle;
}
int minimum = 1 + Math.min(
Math.max(superEggDrop(K - 1, low - 1), superEggDrop(K, N - low)),
Math.max(superEggDrop(K - 1, high - 1), superEggDrop(K, N - high))
);
cache.put(key, minimum);
return cache.get(key);
}
}