887. 鸡蛋掉落

题目

给你 k 枚相同的鸡蛋，并可以使用一栋从第 1 层到第 n 层共有 n 层楼的建筑。

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

每次操作，你可以取一枚没有碎的鸡蛋并把它从任一楼层 x 扔下（满足 1 <= x <= n）。如果鸡蛋碎了，你就不能再次使用它。如果某枚鸡蛋扔下后没有摔碎，则可以在之后的操作中 重复使用 这枚鸡蛋。

请你计算并返回要确定 f 确切的值 的 最小操作次数 是多少？

代码

class Solution {
    int[][] memo;
    public int superEggDrop(int k, int n) {
        memo= new int[k+1][n+1];
        for (int[] m:memo
             ) {
            Arrays.fill(m,-111);
        }
        return dp(k,n);
    }

    int dp(int k,int n){
//        base case
        if (k==1) return n;
        if (n==0) return 0;

        if (memo[k][n]!=-111) return memo[k][n];

        int res =Integer.MAX_VALUE;

//        穷举所有可能选择
//        for (int i = 1; i <n+1 ; i++) {
//            res= Math.min(res, Math.max(dp(k,n-i),dp(k-1,i-1))+1);//碎了或没碎
//        }

        //用二分搜索代替线性搜索
        int lo=1,hi=n;
        while (lo<=hi){
            int mid = (lo+hi)/2;
            int broke = dp(k-1,mid-1);
            int unbroke= dp(k,n-mid);
            if (broke>unbroke) {//broke随mid单增，unbroke单减
                hi=mid-1;
                res=Math.min(res,broke+1);//因为取最坏情况
            }else {
                lo=mid+1;
                res=Math.min(res,unbroke+1);
            }
        }

        //加入备忘录
        memo[k][n]= res;//==Integer.MAX_VALUE? -1:res;
        return res;
    }
}

要点

• 本题利用动归，动归的分支是摔碎了和没摔碎的情况，分别计算出最坏的情况，然后取最小值
• 但这样简单遍历时间复杂度过高，通过二分查找来优化，因为摔坏的的情况，随着楼层i单增，未摔坏的测试次数大小单减。
• 还有第二种方法，利用反向思维，修改 dp 数组的定义，确定当前的鸡蛋个数和最多允许的扔鸡蛋次数，就知道能够确定 F 的最高楼层数，再进行动态规划