题意大致是已知n个鸟蛋和k层楼(均<=1000),最坏情况下至少需要多少次才能测试出鸟蛋的坚硬度。
接触dp以前认为是数学规律,但是没有推出来,后来觉得是搜索,鸟蛋个数为1时,需要试验所有的楼层,当鸟蛋个数大于1时,枚举当前选的楼层,分为破与不破,破了则往下面的楼层继续且鸟蛋数减1,不破就往上面的楼层搜索鸟蛋数不变,然后发现,这不就是dp嘛!
最初始的想法:
dp(i,j)中i代表鸟蛋数,j代表要试验的楼层数,当楼层数为0时自然dp(i,0)也为0,鸟蛋数为1且j不为0时dp(1,j)=j,以此为边界,那么每个dp(i,j)(i>=2,j>=1)都能由前面已知的状态推出。具体递推过程为,枚举本次试验的楼层z,则dp(i,j)=min(1+max(dp(i-1,z-1),dp(i,j-z)))(1<=k<=j),显然i、j均从小到大即可满足所有dp(i,j)都可由已经推出的状态求得。
这里一共有n*k个状态,转移过程需要枚举楼层z数目为k,所以总复杂度应该差不多n*k^2,会不会超时呢?首先,因为题目输入为n,k,假设每次输入都为1000,1000,如果每次输入都执行一遍求解过程显然没有必要,所以很自然的直接求出n=[1,1000],k=[0,1000]时的所有状态,之后O(1)查询即可。同时,即使是这样,最多也依然接近1000*1000*1000。(我以为会超时,但是看不少人似乎这样就过了?不管别人怎么过的,反正我自然就想到了优化...)
由此,我先打出了一部分表,观察规律,最终发现,dp(i,j)要么为dp(i,j-1),要么为dp(i,j-1)+1,因此想到,另开一1005*1005大小的数组m,m(i,x)存的是已经求出的i个鸟蛋时dp值为x的数目,这样就能知道求解dp(i,j)时dp(i,j-1)的数目,假设数目为t,则判断1+max(dp(i,j-1-t),dp(i,t))==dp(i,j-1)是否成立,成立则dp(i,j)=dp(i,j-1),不成立就为dp(i,j)=dp(i,j-1)+1。由此,转移就变成O(1)了,复杂度优化到了n*k。
优化后代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
|