LeetCode887. 鸡蛋掉落

887. 鸡蛋掉落

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

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

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

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

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

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

思路

和蓝桥杯的一题扔手机的差不多

①鸡蛋没碎可以继续用，dp(K,N)表示K个鸡蛋，N层楼的最少移动次数。dp(K,N)=1≤X≤Nmin​(max(dp(K−1,X−1),dp(K,N−X)))

时间复杂度为KN²，超时

②基于最优策略的动态规划，假设x是最优解。dp(K,N)=max(dp(K−1,X−1),dp(K,N−X))，由于dp(K−1,X−1),单调增，dp(K,N−X)单调减，所以当max(dp[k-1][x-1], dp[k][n-x]) > max(dp[k-1][x], dp[k][n-x-1])时x是最优解

思路②代码如下：AC

class Solution {
public:
    int superEggDrop(int K, int N) {
        int dp[N+1];//dp[k-1][n]
        for(int i=0;i<=N;i++)
            dp[i]=i;
        int dp2[N+1];//dp[k][n]
        for(int k=2;k<=K;k++){
            int x=1;
            fill(dp2, dp2+N+1, 0);
            for(int n=1;n<=N;n++){
                while(x<n && (max(dp[x-1], dp2[n-x]) > max(dp[x], dp2[n-x-1])) )
                    x++;
                dp2[n] = 1 + max(dp[x-1], dp2[n-x]);
            }
            for(int i=0;i<=N;i++) dp[i]=dp2[i];
        }
        return dp[N];
    }
};

思路①代码如下：（超时）

class Solution {
public:
    int superEggDrop(int K, int N) {
        int dp[K+1][N+1];
        for(int k=1;k<=K;k++)
        	for(int n=1;n<=N;n++)
            	dp[k][n]=n;
        
        for(int k=2;k<=K;k++){
        	for(int n=1;n<=N;n++){
        		for(int i=1;i<n;i++){
        			int tmp = 1 + max(dp[k-1][i-1], dp[k][n-i]);
        			dp[k][n] = min(dp[k][n], tmp);
				}
			}	
		}
		return dp[K][N];
 	}
};