LeetCode 887 鸡蛋掉落问题:
给你 k 枚相同的鸡蛋,并可以使用一栋从第 1 层到第 n 层共有 n 层楼的建筑。已知存在楼层 f ,满足 0 <= f <= n ,任何从 高于 f 的楼层落下的鸡蛋都会碎,从 f 楼层或比它低的楼层落下的鸡蛋都不会破。
每次操作,你可以取一枚没有碎的鸡蛋并把它从任一楼层 x 扔下(满足 1 <= x <= n)。如果鸡蛋碎了,你就不能再次使用它。如果某枚鸡蛋扔下后没有摔碎,则可以在之后的操作中 重复使用 这枚鸡蛋.请你计算并返回要确定 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 <= k <= 100
1 <= n <= 104
鸡蛋可以重复使用,定义f[i][j]表示用 j 枚鸡蛋最多扔 i 次得到的最大高度,也就是说鸡蛋越多,可以试出的高度约高,也就是f[i][k+1] ≥f[i][k],假设在 i 处,如果鸡蛋碎了,那么可以得到的最大高度为f[i-1][j-1] 如果鸡蛋没碎,那么还可以尝试的高度为 f[i-1][j] 注意是还可以尝试的高度,最终鸡蛋可以确定的楼层的高度为 f[i][j]=f[i-1][j]+f[i-1][j-1]+1;为啥加一呢,就是加入在 i 处碎了,那么该处以下都不会碎,可以确认的最高的高度为 f[i-1][j-1] +1 .
int f[10010][110];
class Solution {
public:
int superEggDrop(int K, int N) {
for(int i=1;i<=N;i++)
{
for(int j=1;j<=K;j++)
{
f[i][j]=f[i-1][j]+f[i-1][j-1]+1;
}
if(f[i][K]>=N) return i;
}
return -1;
}
};
丢旗子问题
一座大楼有层,地面算作第0层,最高的一层为第 层。已知棋子从第0层掉落肯定不会摔碎,从第层掉落可能会摔碎,也可能不会摔碎。给定整数作为楼层数,再给定整数作为棋子数,返回如果想找到棋子不会摔碎的最高层数,即使在最差的情况下扔的最小次数。一次只能扔一个棋子。
示例1
输入:10,1
返回值:10
说明:
因为只有1棵棋子,所以不得不从第1层开始一直试到第10层,在最差的情况下,即第10层是不会摔坏的最高层,最少也要扔10次
示例2
输入:3,2
返回值:2
说明:先在2层扔1棵棋子,如果碎了,试第1层,如果没碎,试第3层
丢旗子问题:由于数据范围很乏,如果使用两重循环会超时,因此需要将两重循环优化为一重循环,状态计算为f[i][j]=f[i-1][j]+f[i-1][j-1]+1;去掉第 i 维,可以得到f [j] =f [j] + f [j-1]+1(等价于 f[i][j]=f[i][j]+f[i][j-1]+1); 但是与原始式子是不等价的,由于需要的是上一个状态的值,要保证上一个状态的值被计算出来,因此需要从大到小循环。
class Solution {
public:
int solve(int N, int K) {
int log2times = log(N)/log(2)+1;
if ( K >= log2times ) return log2times;
int dp[100010] = {0};
for(int i = 1 ; i <= N ; i ++)
{
for(int j = K ; j > 0 ; j --)
{
dp[j] = dp[j - 1] + dp[j] + 1;
if(dp[j] >= N)
return i;
}
}
return 0;
}
};
丢旗子问题:因为牛客网上给定的数据范围(0≤N,K≤106)很大,因此使用上述两层循环会超时,需要使用滚动数组来记录。滚动数组需要保存上一个状态的值即为pre,tem为当前状态的值,当前状态的值为f[i] = f[i] + pre +1 。
class Solution {
public:
int f[10010];
int superEggDrop(int K, int N) {
int log2times = log(N)/log(2)+1;
if ( K >= log2times ) return log2times;
int ret = 0;
int tmp, pre;
while ( true ) {
++ret;
pre = 0;
for ( int i = 0; i < K; ++i ) {
tmp = f[i];
f[i] += pre + 1;
if ( f[i] >= N ) return ret;
pre = tmp;
}
}
return -1;
}
};