题目:
2个鸡蛋,从100层楼上往下扔,以此来测试鸡蛋的硬度,比如鸡蛋在第9层没有摔碎而在第10层摔碎了,那么鸡蛋不会摔碎的零界点就是9层,如何用最少的尝试次数,测试出鸡蛋不会摔碎的临界点?(如果3个鸡蛋呢,如果m个鸡蛋呢)
题目分析:用动态规划,最重要找到状态转移方程,就是写出母问题和子问题之间的关系!比如扔鸡蛋这个问题,你扔碎了,就转化成了往楼上找的子问题;你扔了没碎,那就转化成了往楼下找的子问题。
首先把n层建筑2个鸡蛋所需的最少次数转化为函数dp(2,n)。
2个鸡蛋从第x层扔下所需要的最小次数为dp(2,x),当第1个鸡蛋从第x层扔下时,分为两种情况:
1、第1个鸡蛋碎了,则临界点肯定在1层到x层之间,即转化为x-1层建筑1个鸡蛋所需要的最小次数dp(1,x-1);
2、第1个鸡蛋没碎,则临界点肯定在x+1层到n层之间,即转化为n-x层建筑2个鸡蛋所需要的最小次数dp(2,n-x);
所以,我们在2个鸡蛋x层楼条件下,所需要的最小次数为:dp(2,x)=min( max(dp(1,x-1))+1, dp(2,n-x)+1)。(加1是因为第1个鸡蛋还扔了一次,max是因为只有1个鸡蛋时要考虑最坏情况)
此方法可以扩展到m个鸡蛋,例如3个鸡蛋时,状态转移方程为:dp(3,x)=min( max(dp(2,x-1))+1, dp(3,n-x)+1)。
然后for循环,从第1层到第n层依次填值即可。
//两个鸡蛋100层楼,动态规划
#include<stdlib.h>
#include <stdio.h>
#include <iostream>
#include <vector>
#include <string>
using namespace std;
vector<vector<int> > dp(3,vector<int> (500)); // dp[i][j] 表示用i个鸡蛋测试第 j 层楼所需要的最小次数
void mincount(int n)
{
for (int i=1;i<=n;i++)
dp[1][i]=dp[2][i]=i;
for (int i=1;i<=n;i++)
for (int j=1;j<=i;j++)
dp[2][i]=min(max(dp[1][j-1],dp[2][i-j])+1,dp[2][i]); //dp[2][i]表示用2个鸡蛋,i层楼情况下所需的最小次数。第一次从j层扔下,如果鸡蛋碎了,那么总次数为dp[1][j-1]+1,如果鸡蛋没碎,那么总次数为dp[2][i-j]+1,max是因为每次扔都要考虑最坏情况。
//(该方法可以扩展到多个鸡蛋,如果是3个鸡蛋,递推式为:dp[3][i]=min(max(dp[2][j-1],dp[3][i-j])+1,dp[3][i]))
}
int main()
{
int n;
cin>>n;
mincount(n);
for (int i=1;i<=n;i++)
cout<<dp[1][i]<<" "<<dp[2][i]<<endl;
system("pause");
return 0;
}