lintcode 丢鸡蛋2
描述
有一个n层的建筑。如果一个鸡蛋从第k层及以上落下,它会碎掉。如果从低于这一层的任意层落下,都不会碎。
有m个鸡蛋,用最坏的情况下实验次数最少的方法去找到k, 返回最坏情况下所需的实验次数。
样例
Example 1:
Input: m = 2, n = 100
Output: 14
Example 2:
Input: m = 2, n = 36
Output: 8
思路
利用动态规划,dp[i][j]表示i个鸡蛋试验j层楼。
对于任意一次dp[i][j],第一次尝试在k层,那么dp[i][j] = 1+max(dp[ i ][ j - k ], dp[ i - 1 ][ k - 1]).
找最大值是考虑到最坏情况,要找的k是最优k,即要让dp[i][j]最小。
代码
class Solution {
public:
/**
* @param m: the number of eggs
* @param n: the number of floors
* @return: the number of drops in the worst case
*/
int dropEggs2(int m, int n) {
// write your code here
int dp[m+1][n+1];
for (int i = 0; i < m+1; i++) {
for (int j = 0; j < n+1; j++)
dp[i][j] = INT_MAX;
}
for (int i = 0; i < n+1; i++) {
dp[1][i] = i;
dp[0][i] = 0;
}
for (int i = 0; i < m+1; i++) {
dp[i][0] = 0;
dp[i][1] = 1;
dp[i][2] = 2;
}
for (int i = 2; i < m+1; i++) {
for (int j = 2; j < n+1; j++) {
for (int k = 2; k < n; k++) {
dp[i][j] = min(dp[i][j], 1+max(dp[i][j-k], dp[i-1][k-1]));
}
}
}
return dp[m][n];
}
};