【Leetcode】1884. Egg Drop With 2 Eggs and N Floors

题目地址：

https://leetcode.com/problems/egg-drop-with-2-eggs-and-n-floors/description/

给定两个鸡蛋，再给定$n$层高的楼，每一层编号$1,2,...,n$。已知存在一个楼层$x,0\le x\le n$，在该楼层扔鸡蛋不会碎，但在其上面的楼层扔鸡蛋则会碎。同一个鸡蛋如果不碎，可以扔多次。问最少扔多少次能确定$x$。

法1：动态规划。设$f[k][i]$是一共有$k$个鸡蛋$i$层楼的情况下，最少扔多少次能确定$x$。如果$k=1$，那唯一的办法就是从第$1$层开始依次增大楼层扔，一旦发现碎了，那答案就是碎了的那一层编号减$1$，至少扔$n$次才能确定答案。接下来考虑$k=2$，可以枚举第$1$个鸡蛋扔的楼层，如果第$1$个鸡蛋在$y$层扔的，分两种情况，如果碎了，那么接下来至少需要扔$f[1][y-1]$次，总共次数为$1+f[1][y-1]$；如果没碎，则还是有两个鸡蛋，但是楼层的范围缩小为$y+1,...,i$，所以至少还需要扔$f[2][i-y]$次，总次数为$1+f[2][i-y]$。所以要用 max ⁡ { f [ 1 ] [ y − 1 ] + 1 , f [ 2 ] [ i − y ] + 1 } \max\{f[1][y-1]+1,f[2][i-y]+1\} max{f[1][y−1]+1,f[2][i−y]+1}来更新$f[2][i]$。最后返回$f[2][n]$即可。代码如下：

class Solution {
 public:
  int twoEggDrop(int n) {
    int f[3][n + 1];
    memset(f, 0x3f, sizeof f);
    f[1][0] = f[2][0] = 0;
    for (int i = 1; i <= n; i++) f[1][i] = i;
    for (int i = 1; i <= n; i++)
      for (int j = 1; j <= i; j++)
		f[2][i] = min(f[2][i], 1 + max(f[1][j - 1], f[2][i - j]));
    return f[2][n];
  }
};

时间复杂度$O(n^2)$，空间$O(n)$。

法2：参考https://blog.csdn.net/qq_46105170/article/details/118288285。设$f[i][j]$是在有$j$个鸡蛋且总共允许扔$i$次的情况下，能确定求出临界点$x$的大楼的最多层数，则$f[i][j]=f[i-1][j-1]+1+f[i-1][j]$。本题中令$j=2$，那么$f[i][2]=f[i-1][1]+1+f[i-1][2]$，而$f[i][1]=i$，所以$f[i][2]=i+f[i-1][2]$，所以$$f[i][2]=i+(i-1)+...+1$$找到$f[i][2]\ge n$最小的$i$即可。代码如下：

class Solution {
 public:
  int twoEggDrop(int n) {
    int res = 1, tot = 1;
    while (tot < n) tot += ++res;
    return res;
  }
};

时间复杂度$O(\sqrt{n})$，空间$O(1)$。