面试经典题-鸡蛋掉落

题目来源：https://leetcode-cn.com/problems/super-egg-drop/

你将获得 K 个鸡蛋，并可以使用一栋从 1 到 N  共有 N 层楼的建筑。

每个蛋的功能都是一样的，如果一个蛋碎了，你就不能再把它掉下去。

你知道存在楼层 F ，满足 0 <= F <= N 任何从高于 F 的楼层落下的鸡蛋都会碎，从 F 楼层或比它低的楼层落下的鸡蛋都不会破。

每次移动，你可以取一个鸡蛋（如果你有完整的鸡蛋）并把它从任一楼层 X 扔下（满足 1 <= X <= N）。

你的目标是确切地知道 F 的值是多少。

无论 F 的初始值如何，你确定 F 的值的最小移动次数是多少？

这题显然是个动态规划问题，重点在于怎么定义状态和状态转移方程。

一种比较直观简洁的算法：

• 定义状态dp[i][j]：i个蛋、扔j次，能够确定的最大层数。
• 状态转移方程：任意选一层x去丢，如果蛋没碎，那么可以继续往上确定dp[i][j-1]层；如果蛋碎了，那么也就是说这一次扔蛋没有带来额外的有用线索，此时能确定的层数等同于i-1个蛋扔j-1次时的结果，即dp[i-1][j-1]层；此外，扔的这一次如果没碎，那么本层是安全的。于是有dp[i][j] = dp[i][j-1] + (dp[i-1][j-1] + 1)

class Solution(object):
    def superEggDrop(self, K, N):
        """
        :type K: int
        :type N: int
        :rtype: int
        """
        dp = [[0] * (N+1) for i in range(K+1)]      # dp[i][j]表示i个蛋，丢j次能确定的最大层数
        for i in range(1, K+1):
            for j in range(1, N+1):
                dp[i][j] = dp[i][j-1] + dp[i-1][j-1] + 1
                if dp[K][j] >= N:         # 一旦找到一个K个蛋能确定N层的状态，就返回对应的次数j
                    return j
        return 0

其他方法待添加。。。

 

参考：

http://www.freesion.com/article/588062988/