887.鸡蛋掉落

887.鸡蛋掉落

你将获得 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 是多少。

示例 2：

输入：K = 2, N = 6
输出：3

示例 3：

输入：K = 3, N = 14
输出：4

提示：

1. 1 <= K <= 100
2. 1 <= N <= 10000

使用动态规划算法求解

问题状态包括剩余鸡蛋数目和需要测试的楼层，用 (K, [N]) 表示

当在第 i 层扔鸡蛋时，有两个结果：

• 鸡蛋碎了
• 鸡蛋没碎

若鸡蛋碎了，可以确定高于 i 楼扔鸡蛋鸡蛋都会碎，此时还剩 K-1 个鸡蛋，需要继续判断 1 至 i-1 楼共 i-1 层楼的情况；(K, N) -> (K-1, i-1)

若鸡蛋没碎，可以确定低于 i 楼扔鸡蛋鸡蛋都不会碎，此时还剩 K 个鸡蛋，需要继续判断 i+1 至 N 楼共 N-i 层楼的情况；(K, N) -> (K, N-i)

当只有一个鸡蛋时（K = 1），需要自底向上扔，最坏情况下需要扔 N 次才能知道高于某个楼层鸡蛋都会碎。

class Solution:
    def superEggDrop(self, K: int, N: int) -> int:
        
        memo = {}
        def dp(K, N):
            """返回确定知道高于某个楼层扔鸡蛋鸡蛋会碎的尝试次数。
            """
            if K == 1: return N
            if N == 0: return 0
            if (K, N) in memo:
                return memo[(K, N)]
            
            res = float('inf')

            lo = 1
            hi = N
            while lo <= hi:
                mid = (lo + hi) // 2
                # 鸡蛋碎了
                broken = dp(K-1, mid-1)
                # 鸡蛋没碎
                not_broken = dp(K, N-mid)
                if broken > not_broken:
                    hi = mid - 1
                    res = min(res, broken+1)
                else:
                    lo = mid + 1
                    res = min(res, not_broken+1)

            memo[(K, N)] = res
            return res
        return dp(K, N)