Leetcode 鸡蛋掉落
Leetcode 鸡蛋掉落
你将获得 K 个鸡蛋,并可以使用一栋从 1 到 N 共有 N 层楼的建筑。
每个蛋的功能都是一样的,如果一个蛋碎了,你就不能再把它掉下去。
你知道存在楼层 F ,满足 0 <= F <= N 任何从高于 F 的楼层落下的鸡蛋都会碎,从 F 楼层或比它低的楼层落下的鸡蛋都不会破。
每次移动,你可以取一个鸡蛋(如果你有完整的鸡蛋)并把它从任一楼层 X 扔下(满足 1 <= X <= N)。
你的目标是确切地知道 F 的值是多少。
无论 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
这道题还是很有意思的
初读题会觉得就是一个简单的数学奥赛问题,但是仔细读题,发现问的是:
因此,在最坏的情况下我们需要移动 2 次以确定 F 是多少。
就是说在鸡蛋都用完的情况下,怎么移动的次数是最少的。
思路
联想到二分法,比如说100层楼两个蛋,第一个在50层扔了,如果没摔碎,那么在50之上继续做。
如果摔碎了,就不能浪,必须从第一层一层层往上摔。
/*
若测试k次,能够确定结果的最大测量楼层范围为[0,dp[k][m]],则称dp[k][m]为最大测量楼层,则问题就是就是找到一个最小的m使得dp[k][m]>=N
转移方程dp[k][m] = dp[k-1][m-1] + dp[k][m-1] + 1
原因:假设第一次是在X层测试,
若碎了,则F<X,需要测试它下面的楼层,蛋的数量和次数减少1,它的最大测量楼层为dp[k-1][m-1],还需要测试的楼层范围为[0,X-1]
若没碎,则F>=X,需要测试它上面的楼层,蛋的数量不变,次数减少1,它的最大测量楼层为dp[k][m-1],还需要测试的楼层范围为[X,dp[k][m]]
若想达到最大测量层数,则分开的两个区间的长度都应该达到最大测量楼层+1。即X-1 = dp[k-1][m-1],dp[k][m] - X = dp[k][m-1]
化去X,得到dp[k][m] = dp[k-1][m-1] + dp[k][m-1] + 1
注意左侧都是m,右侧都是m-1,可以只使用一列存储数据,K从大到小进行更新
*/
代码
class Solution {
/*
若测试k次,能够确定结果的最大测量楼层范围为[0,dp[k][m]],则称dp[k][m]为最大测量楼层,则问题就是就是找到一个最小的m使得dp[k][m]>=N
转移方程dp[k][m] = dp[k-1][m-1] + dp[k][m-1] + 1
原因:假设第一次是在X层测试,
若碎了,则F<X,需要测试它下面的楼层,蛋的数量和次数减少1,它的最大测量楼层为dp[k-1][m-1],还需要测试的楼层范围为[0,X-1]
若没碎,则F>=X,需要测试它上面的楼层,蛋的数量不变,次数减少1,它的最大测量楼层为dp[k][m-1],还需要测试的楼层范围为[X,dp[k][m]]
若想达到最大测量层数,则分开的两个区间的长度都应该达到最大测量楼层+1。即X-1 = dp[k-1][m-1],dp[k][m] - X = dp[k][m-1]
化去X,得到dp[k][m] = dp[k-1][m-1] + dp[k][m-1] + 1
注意左侧都是m,右侧都是m-1,可以只使用一列存储数据,K从大到小进行更新
*/
public int superEggDrop(int K, int N) {
int[] dp = new int[K+1]; //int型会自动初始化为0
int m = 0;
//从底层开始找
while (dp[K]<N){
m++;
for (int i = K; i >= 1; i--)
{
dp[i] = dp[i] + dp[i-1] +1;
}
}
return m;
}
}
leetcode上最快的代码:
class Solution {
public int superEggDrop(int K, int N) {
int[] d=new int[K+1];
for (int i=0;i<d.length;i++){
d[i]=1;
}
//当鸡蛋数为0时,能确定的楼层数也就是0
d[0]=0;
//扔鸡蛋的次数
int x=1;
//当K个鸡蛋确定的楼层数大于等于N时,最少移动次数也就找到了
while (d[K]<N){
for (int i=K;i>=1;i--){
//i代表鸡蛋数
d[i]=d[i]+d[i-1]+1;
}
x++;
}
return x;
}
}