LeetCode 887. 鸡蛋掉落（二分优化dp）

题意：

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

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

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

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

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

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

 
数据范围:
1 <= K <= 100
1 <= N <= 10000

解法：

d[i][j]表示i楼,j个鸡蛋的最坏情况下的最小移动次数.

对第j个鸡蛋,在第t层丢,如果碎了,那么接下来变为d[t-1][j-1],
如果没碎,那么前t层不用测了,只需要测剩下i-t层,那么接下来变为d[i-t][j].
因为要考虑最坏情况,所以要对碎和不碎取max,
因为要求最小次数,所以对所有t的情况取min.

这样的复杂度是O(n*k*n)的,会超时,需要优化.

在枚举t的过程中,
d[t-1][j-1]是递增的,
d[i-t][j]是递减的,
因此max(d[t-1][j-1],d[i-t][j])的图像是单峰的,如下图.

这样的图可以三分求峰值,
同时这种max(f(i),g(i))的单峰函数,好像可以直接二分:
二分mid,如果f(mid)>g(mid),那么说明mid在峰值左边,否则在右边.

优化之后复杂度就优化为O(n*k*log)了.

code：

class Solution {
public:
    int superEggDrop(int k, int n) {
        vector<vector<int> >d(n+5,vector<int>(k+5,0));
        for(int i=1;i<=n;i++){
            d[i][0]=1e9;
            for(int j=1;j<=k;j++){
                d[i][j]=1e9;
                int l=1,r=i;
                while(r-l>3){
                    int mid=(l+r)/2;
                    if(d[i-mid][j]>=d[mid-1][j-1]){
                        l=mid;
                    }else{
                        r=mid;
                    }
                }
                for(int t=l;t<=r;t++){
                    d[i][j]=min(d[i][j],max(d[t-1][j-1],d[i-t][j])+1);
                }
            }
        }
        return d[n][k];
    }
};

---