【基础算法】切割钢管与动态规划
尽管排序算法还有很多没有说,但因为这篇文章是已经现成有的,就先上这个,回头再把排序补一下。
好的开始~BigMoyan有一个好基友叫zou先生,zou先生除了是BigMoyan在学校的社团老大外,还是一家专门为夜总会提供钢管的公司的区域经理。最近,zou经理发现这样一个事情,夜总会需要各种长度的钢管用作各种用途,然而每种长度的钢管的价格却不一样,总而言之如下表。
从前,zou经理总是傻乎乎的把总公司发来的10m长钢管以25块的价格卖出去,但是某天他跟学校的打菜阿姨聊起人生和事业的时候,打菜阿姨却来了一句:
“明明每根钢管能卖27块钱,你只卖25,学过C++吗?”
这深深伤害了zou经理的心灵,于是他向室友Jie求救。
以下是BigMoyan转述Jie的分析:
首先我们正式的表达一下这个问题,有一段长度为i的钢管,整段出售的价格为Pi,求适当的切割钢条方案使得获利最大。为了便于zou理解问题,jie先用一个例子做了一下简单说明。
比如你现在有一段长为4m的钢管,如果整段卖的话,卖9块。然而你也可以切成两段2m的钢管分别卖,这样一共可以卖5*2=10块,所以并不是整段卖就一定获益最大的。假设对于长为i的钢管,获益最大的售价为Ri,我们试着穷举一下。
长为1的钢管,i=1,只能整段出售,R1=1
长为2的钢管,i=2,很容易发现整段出售获益最高,R2=2
…
长为4的钢管,i=4,上面分析了应该切成两段,R4=10=5+5
长为5的钢管,整段出售是10块,分成1+4,因为R4=10,所以可以卖11块,分成2+3,则可卖13块,因此R5=2+3=13
…
由于jie看了BigMoyan关于快速排序的介绍,受到分治策略的影响,于是jie刚开始自然而然的认为这是一个可以递归解决的问题,对于长为i的钢管,其最大获益为:
Ri= max{Pn, R1+Rn-1,R2+Rn-2,…Ri-1+R1}
这个意思是说,为了求Ri,我们可以把钢管分为和为i的两段,而这两段各自又是一个求大获益的问题,正如BigMoyan曾经说过的那样,想要理解递归,首先,你要理解递归。然后在各种分法中找出最大的那个,自然就是结果。注意上式中第一个参数Pn为整段出售的价格。
Jie把思路告诉了BigMoyan,BigMoyan很快为zou写了这样一段程序(没错我就是这样一个仗义的人),用C++写大概是下面这个样子(BigMoyan用的最多的依然是Python,可惜Python没有做尾递归优化,只能暂时用一把C++了):
#include<iostream> using namespace std; int Max(int i, int j){ return i>j?i:j; } int cut_rod(int p[],int n){ if(n==0) return 0; int f=0; int q=-1; for(int i=1;i!=n;i++){ f=Max(p[n-1],cut_rod(p,i)+cut_rod(p,n-i)); q=Max(q,f); } return q; } int main(){ int p[10]={1, 5, 8, 9, 10, 17, 17, 20, 24, 25}; int f=cut_rod(p,10); cout<<f; return 1; }
老规矩,碰到代码不要慌,拿个实例稍微分析一下,例如求长为5的钢管的最大收益。调用函数cut_rod(p,5)
进去以后, if测试失败,进入for循环
i=1:
f=Max(p[4],cut_rod(p,1)+cut_rod(p,4));
找出了p[4]和cut_rod(p,1)+cut_rod(p,4)的大者,前者为5m钢管整段出售的价格(下标从0开始所以是5-1=4),后者将5m分为1m和4m,将两段的最大收益加起来作为进行1+4分割的最大收益。
q=Max(q,f);
第一趟的时候q=-1,所以此时q=f保存1+4分割时的最大收益。
i=2:
f=Max(p[4],cut_rod(p,1)+cut_rod(p,4));
如上,此次找的是整段出售和2+3分割的最大收益之间的大者。
q=Max(q,f);
q在上次保存了f,这次将新f和上一次的值(q)比较,拿出最大的作为最大收益
后面以此类推
然而在测试代码的时候BigMoyan发现,这代码执行效率太他喵的低了!n比较大的时候,基本上n增加1,代码运行时间就要加两倍,我要敢测试一个500m长的钢管的最大收益(假设数据都有),估计zou经理都要毕业了。
稍加分析发现,执行效率低的原因在于子问题一直在重复计算!计算10m钢管收益的时候要把1~9m的收益全算一次,而为了计算9m钢管的收益,我又要把1~8m的都算一次。
心!好!累!
此时聪明的jie也已经察觉到了问题所在,这个问题看起来是递归,其实却与递归略有不同。递归的子问题互相不相交,而这个东西的子问题是相交的,相同的子问题被一遍一遍的计算,才导致了效率的低下。
Jie略一思忖,计上心来,既然子问题被一遍遍的计算,我们何不以空间换时间,把已经被计算好的子问题的结果保存起来,当需要的时候首先查询,如果这个问题已经算好了,直接拿来用就是,若没算好再进行计算。
BigMoyan拍案叫绝,再次改写了代码,于是就有了下面这段C++ code:
nt cut_rod(int p[],int n){ int *r=new int[n]; for(int i=0;i!=n;i++) r[i]=-1; return cut_rod_memo(p,n,r); }
调用函数名保持一致方便理解,首先建立一个备忘录r,r用来保存已经被计算好的子问题的答案。将它初始化为-1。我们还是以计算5m钢管为例,那么r被初始化为[-1,-1,-1,-1,-1].
接着调用cut_rod_memo来计算最大收益,那么cut_rod_memo是什么呢?
nt cut_rod_memo(int p[], int n, int r[]){ if(r[n-1]>=0) return r[n-1]; if(n==0) int q=0; else{ int f=0; int q=-1; for(int i=1;i!=n;i++){ f=Max(p[n-1],cut_rod_memo(p,i,r)+cut_rod_memo(p,n-i,r)); q=Max(q,f); } r[n-1]=q; return q; } }
好的,不要头大,BigMoyan慢慢分析。
进入cut_rod_memo后,首先查询5m的结果是不是已经算好了,答案是当然没有,r[4]=-1。于是进入else语句,这里的语句跟之前递归的版本一模一样,区别只在于递归调用的是cut_rod_memo,因为该函数里第一个语句就是判断子问题答案有没有做好,所以避免了多次计算相同的子问题,下面展示这一过程。
i=1:
由之前的无脑递归的版本可知,for循环解决了i=1时的最大收益,在for循环完成后,将其保存在r[0]中,此时r[0]=1
i=2:
在递归调用cut_rod_memo(p,1,r)时,因为已经算好r[0]=1,故cut_rod_memo的第一个if判断成功,返回r[0],此时的返回值由已经计算过的结果直接拿出,没有重新计算。
其他情况同理。
BigMoyan与jie两人相对而笑,笑容中充满着对对方的赞赏。他们一起来到zou经理折戟的银桦食堂,与打菜阿姨如此这般一讲,本指望收到阿姨赞美的眼光,没想到阿姨把勺子里的菜抖掉一些,淡然说道:
“唔?这不就是动态规划吗,什么新鲜东西,值得来浪,刷卡!”
Ps.动态规划问题当然不止这一类,但其基本思想是一致的,就是不要重复解决已经解决过的问题,感兴趣的同学可以百度一下动态规划
PPS:下面是全部C++代码
1 #include<iostream> 2 using namespace std; 3 4 int Max(int i, int j){ 5 return i>j?i:j; 6 } 7 8 int cut_rod_memo(int p[], int n, int r[]){ 9 if(r[n-1]>=0) 10 return r[n-1]; 11 if(n==0) 12 int q=0; 13 else{ 14 int f=0; 15 int q=-1; 16 for(int i=1;i!=n;i++){ 17 f=Max(p[n-1],cut_rod_memo(p,i,r)+cut_rod_memo(p,n-i,r)); 18 q=Max(q,f); 19 } 20 r[n-1]=q; 21 return q; 22 } 23 } 24 25 int cut_rod(int p[],int n){ 26 int *r=new int[n]; 27 for(int i=0;i!=n;i++) 28 r[i]=-1; 29 return cut_rod_memo(p,n,r); 30 } 31 32 int main(){ 33 int p[10]={1, 5, 8, 9, 10, 17, 17, 20, 24, 25}; 34 int f=cut_rod(p,10); 35 cout<<f; 36 return 1; 37 }
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。