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];
}
};