1884. 鸡蛋掉落-两枚鸡蛋
难度中等28
给你 2 枚相同 的鸡蛋,和一栋从第
1
层到第n
层共有n
层楼的建筑。已知存在楼层
f
,满足0 <= f <= n
,任何从 高于f
的楼层落下的鸡蛋都 会碎 ,从f
楼层或比它低 的楼层落下的鸡蛋都 不会碎 。每次操作,你可以取一枚 没有碎 的鸡蛋并把它从任一楼层
x
扔下(满足1 <= x <= n
)。如果鸡蛋碎了,你就不能再次使用它。如果某枚鸡蛋扔下后没有摔碎,则可以在之后的操作中 重复使用 这枚鸡蛋。请你计算并返回要确定
f
确切的值 的 最小操作次数 是多少?
示例 1:
输入:n = 2
输出:2
解释:我们可以将第一枚鸡蛋从 1 楼扔下,然后将第二枚从 2 楼扔下。
如果第一枚鸡蛋碎了,可知 f = 0;
如果第二枚鸡蛋碎了,但第一枚没碎,可知 f = 1;
否则,当两个鸡蛋都没碎时,可知 f = 2。
示例 2:
输入:n = 100
输出:14
解释:
一种最优的策略是:
- 将第一枚鸡蛋从 9 楼扔下。如果碎了,那么 f 在 0 和 8 之间。将第二枚从 1 楼扔下,然后每扔一次上一层楼,在 8 次内找到 f 。总操作次数 = 1 + 8 = 9 。
- 如果第一枚鸡蛋没有碎,那么再把第一枚鸡蛋从 22 层扔下。如果碎了,那么 f 在 9 和 21 之间。将第二枚鸡蛋从 10 楼扔下,然后每扔一次上一层楼,在 12 次内找到 f 。总操作次数 = 2 + 12 = 14 。
- 如果第一枚鸡蛋没有再次碎掉,则按照类似的方法从 34, 45, 55, 64, 72, 79, 85, 90, 94, 97, 99 和 100 楼分别扔下第一枚鸡蛋。
不管结果如何,最多需要扔 14 次来确定 f 。
解析:
-
dp定义
dp(n, k)
表示现在有n
层楼需要验证,此时你手里有k
个鸡蛋,返回此时的最小操作次数。 -
base case
-
如果没有楼层需要验证了(n == 0),那你不管有几个鸡蛋都不用操作了。
-
如果你手里只有一个鸡蛋了(k == 1),那你只能从1楼开始试:1楼、2楼、…、n楼,共需要n次操作。
if(n == 0){ return 0; } if(k == 1){ return n; }
-
-
状态转移
现在手里有k个鸡蛋,接下来选一层扔鸡蛋,假设选了第i层,
有两种情况:
1:碎了,则要确定的f在[0,i-1]内,则需要验证的楼层未i-1,除此之外损失了一个鸡蛋,只剩下k-1个鸡蛋,搜索的楼层区间应该从[1..N]
变为[1..i-1]
共i-1
层楼;,即接下来验证dp[i-1,k-1],操作次数+1;
2:没碎:则要确定的f在[i,n]内,接下来需要验证的楼层数为n-i,并且没有损失鸡蛋,搜索的楼层区间应该从 [1..N]
变为[i+1..N]
共N-i
层楼。即接下来验证dp[n-i,k],操作次数+1;
题目要求是:最坏情况下最少需要操作几次,最坏情况下在这两种情况之间,选择那个需要操作次数最大的,再加上自己扔鸡蛋的操作,即Math.max(dp[i-1,k-1],dp[n-i,k])+1,这个值就是你选择从第i层楼扔鸡蛋,最坏情况下需要的操作次数
-
dp[n,k]即为需要的操作次数,
是你在所有楼层中选择一个最坏情况下需要的操作次数最小的楼层扔鸡蛋。所以需要遍历所有楼层,找到Math.min(res, Math.max(dp(i-1, k-1), dp(n-i, k)) + 1)。
-
添加一个备忘录消除重叠子问题即可
代码:
class Solution {
int[][] memo;
public int twoEggDrop(int n) {
memo=new int[n+1][3];
return dp(n,2);
}
public int dp(int n,int k){
//1. basecase
if(n==0) return 0;
if(k==1) return n;
//2. 查表
if(memo[n][k]!=0) return memo[n][k];
//3.状态转移
int res=Integer.MAX_VALUE;
for(int i=1;i<=n;i++){
res=Math.min(res,Math.max(dp(i-1,k-1),dp(n-i,k))+1);
}
memo[n][k]=res;
return res;
}
}