题目描述:
你将获得 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 是多少。
解法一
思路分析:
假设dp[M][K]为K个鸡蛋移动M步,代表最坏情况下移动的次数,则动态方程可以设为
dp[m][k]=dp[m-1][k-1]+dp[m-1][k]+1
如果鸡蛋碎了,那么就检查dp[m-1][k-1]层,否则就检查dp[m-1][k]层
时间复杂度:O(NK),空间复杂度O(KlogN):
python版:
def superEggDrop(self, K, N):
dp = [[0] * (K + 1) for i in range(N + 1)]
for m in range(1, N + 1):
for k in range(1, K + 1):
dp[m][k] = dp[m - 1][k - 1] + dp[m - 1][k] + 1
if dp[m][K] >= N: return m
解法二:
将DP转换为1P问题:
js版本:
时间复杂度O(klogN),空间复杂度O(K)
/**
* @param {number} K
* @param {number} N
* @return {number}
*/
var superEggDrop = function(K, N) {
let dp=new Array(K+1).fill(0)
let count=0;
while(dp[K]<N){
count++
for(let k=K;k>0;k--){
dp[k]+=dp[k-1]+1
}
}
return count
};
python版本:
时间复杂度:O(min(K,logN)^2),空间复杂度O(min(K,logN)
class Solution(object):
def superEggDrop(self, K, N):
"""
:type K: int
:type N: int
:rtype: int
"""
dp = [0, 0]
m = 0
while dp[-1] < N:
for i in range(len(dp) - 1, 0, - 1):
dp[i] += dp[i - 1] + 1
if len(dp) < K + 1:
dp.append(dp[-1])
m += 1
print(dp)
return m