题意:有 k k k个鸡蛋, n n n层高的楼,问:最坏情况下,最少需要扔多少次鸡蛋才能知道从第几层楼开始往下扔鸡蛋,鸡蛋会碎。
思路:这个题原题题意太迷,我还说二分呢,后来认真看看题目才知道,鸡蛋是有限的,也就是说,假设我们有两个鸡蛋,要测100层楼,比如我从50层先扔下去一个,它碎了,那我们另一个就不能冒险,只能从第一层开始,假设在第49层碎了,那扔了50次才试出来。
动态规划的思想可以很好的解决这个问题,假设
f
[
i
]
[
j
]
f[i][j]
f[i][j]表示使用
i
i
i个鸡蛋,扔
j
j
j次,准确测出的楼层数,若准确测出的楼层数大于
n
n
n,则答案就是
j
j
j,这个准确测出的意思就是比如我们有1个鸡蛋,我们只能准确测出1层,因为在第1层扔了,如果碎了,那就是0层,没碎就是1层。
我们现在的局面是 i i i个鸡蛋扔 j j j次,这个局面可以由前两种局面转化而来,第一种是鸡蛋没碎,也就是 i i i个鸡蛋扔了 j − 1 j-1 j−1次,第二个是鸡蛋碎了,也就是 i − 1 i-1 i−1个鸡蛋扔了 j − 1 j-1 j−1次。
转移方程就是:
f
[
i
]
[
j
]
=
f
[
i
−
1
]
[
j
−
i
]
+
f
[
i
]
[
j
−
1
]
+
1
f[i][j]=f[i-1][j-i]+f[i][j-1]+1
f[i][j]=f[i−1][j−i]+f[i][j−1]+1
代码就很明显了。
class Solution {
public int superEggDrop(int K, int N) {
int[][] dp = new int[K + 1][N + 1];
for (int k = 1; k <= K; k++) {
for (int m = 1; m <= N; m++) {
dp[k][m] = dp[k][m - 1] + dp[k - 1][m - 1] + 1;
if (dp[k][m] >= N) {
return m;
}
}
}
return N;
}
}