你将获得 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