【算法导论】动态规划之“钢管切割”问题
动态规划,其实跟分治法有些相似,基本思想都是将复杂的问题分成数个简单的子问题,然后再去解决。它们的区别在于,分治法关注的子问题不相互“重叠”,而动态规划关注的子问题,多是相互“重叠”的。比如在快速排序中,我们将数据分成两部分,这两部分再分别快速排序的递归思想,也就是将整个问题的排序划分为子问题子数组的排序,但是这两个子数组的排序之间并没有相互联系,a子数组的排序不会因为b子数组的排序而得到任何“好处”或者“坏处”。但是有些时候,划分的子问题之间却是有联系的,比如下面的“钢管切割”问题:
钢管切割原始问题:
某公司生产长钢管,然后一般,会将钢条切断,变成不同长度,然后去售卖。其中有个问题是,不同长度的钢管的售价是不一样的,但是它们并不是完全按照比例来,比如2米的钢管售价要比3米的钢管售价要少,但是并不是2比3的比例。钢管的长度售价表如下:
长度i | 1 2 3 4 5 6 7 8 9 10 |
价格Pi | 1 5 8 9 10 17 17 20 24 30 |
于是问题就来了,比如30米长的钢管,要如何切割,切割成多长的几条,才能让售价最高,收益最高呢?
求解最佳收益和对应的分配方法:
朴素算法:
最简单直接的想法,就是用暴力破解,n长的钢管,可以分解成i长和n-i长的两段,因为i可以从0~n取值,所以我们可以对i不进行继续切割,于是对于长为i的这一段,可以直接调用价钱数组p[i]来得到价钱,然后加上对n-i递归调用求最优收益的函数的返回值。在过程之中记录这些组合的最优收益,等循环结束的时候,就能得到最优的收益价钱。
假设r[n]代表的是n长的钢管的切割最佳收益值,数组p代表上面表中的价格,其中p[0]=0,从p[1]~p[10]对应上面表中的数据,那么按照上面的想法,有公式:
r[n]=max(p[i],r[n-i]),i从1到n,当n=0时,r[n]=0,因为0长的钢管售价当然为0。
于是给以下实现代码:
int cut_rod(int* p, int n) { if (n == 0) { return 0; } int q = -1; for (int i = 1; i <= n; i++) { /* * 将n长的钢条,分成i和n-i的两段,i长的那段不切割,而n-i的那段求最大 * 切割收益方式,然后相加;而q值是所有的组合中,最大收益的那个 */ q = max(q, p[i] + cut_rod(p, n - i)); } return q; }
这种方法比较容易理解,但是性能是不是好呢?
可以简单的以n=4的情况来看一下:
n=4的划分(其中前面的那一段是直接使用p[i],后面一段调用函数来求最佳收益):
cut_rod(p,4)的划分可能:
①1长和3长:p[1]+cut_rod(p,3)
②2长和2长:p[2]+cut_rod(p,2)
③3长和1长:p[3]+cut_rod(p,1)
④4长和0长:p[4]+cut_rod(p,0)
而其中cut_rod(p,3)又可以划分为数组p中元素与cut_rod(p,0),cut_rod(p,1)和cut_rod(p,2);以此类推,可以给出一种以递归调用树的形式展示cut_rod递归调用了多少次:
不难从图中看出,做了大量重复工作,以n=2的节点为例,分别在n=4和n=3的时候都被调用了。根据上图,可以给出递归调用次数的一个公式,假设T(n)表示cut_rod第二个参数为n时的调用次数,T(0)这时候是为1的,因为根结点的第一次调用也要算进去。于是有:
T(n)=1+T(0)+T(1)+...+T(n-1)
使用归纳法,可以比较容易的得出:T(n)=2^n。
指数次幂的调用次数,显然太大,我们稍微让n大一点,则会让整个过程变的漫长。
动态规划算法:
而实际上我们不需要在每次都去重新计算cut_rod的在n=2时的结果,只需要在第一次计算的时候将结果保存起来,然后再需要的时候直接使用即可。这其实就是所谓的动态规划算法。
这里的思路有两种,一种叫带备忘的自顶向下方法,是顺着之前的代码,当需要的时候去检查是不是已经计算好了,如果是,则直接使用,如果不是,则计算,并保存结果。第二种思路是自底向上方法,不论需不需要,先将子问题一一解决,然后再来解决更一级的问题,但要注意的是,我们需要先从最小的子问题开始,依次增加规模,这样每一次解决问题的时候,它的子问题都已经计算好了,直接使用即可。
带备忘的自顶向下方法:
int memoized_cut_rod_aux(int* p, int n, int* r) { if (r[n] >= 0) { return r[n]; } int q = -1; if (n == 0) { q = 0; } else { for (int i = 1; i <= n; i++) { q = max(q, p[i] + memoized_cut_rod_aux(p, n - i, r)); } } r[n] = q; return q; } /* * 自顶向上的cut-rod的过程 */ int memoized_cut_rod(int* p, int n) { int* r = new int[n + 1]; //初始化r数组,r数组用来存放,某种解决方案的最大收益值,对于n长的钢条而言,有n+1种切割方案,所以数组n+1长 for (int i = 0; i <= n; i++) { r[i] = -1; } return memoized_cut_rod_aux(p, n, r); }
自底向上的方法:
/* * 自底向上的方式,先计算更小的子问题,然后再算较大的子问题,由于较大的子问题依赖于更小的子问题的答案,所以在计算较 * 大的子问题的时候,就无需再去计算更小的子问题,因为那答案已经计算好,且存储起来了 */ int bottom_up_cut_rod(int p[], int n) { int* r = new int[n + 1]; r[0] = 0; //将r[0]初始化为0,是因为0长的钢条没有收益 for (int j = 1; j <= n; j++) { int q = -1; /* * 这里不用i=0开始,因为i=0开始不合适,因为这里总长就是为j,而划分是i和j-i的划分,如果i等于0,那么 * 就意味着要知道r[j-0]=r[j]的值也就是j长的最好划分的收益,但是我们这里不知道。而且对于p[0]而言本身就没有意义 * p数组中有意义的数据下标是从1到n的 */ for (int i = 1; i <= j; i++) { q = max(q, p[i] + r[j - i]); // } r[j] = q; } return r[n]; }
上面两种算法的时间复杂度都是O(n^2)。
重构解
上面的代码只给出了最优的收益值,但是却没有给出最优收益到底是在那种切割分配方式下得到的,比如说n=9时,最佳收益为25,要分成3和6两段。这里可以使用另一个数组s来存储分段情况,比如s[9]存储3,然后我们让n=9-3,就可以得到s[6]的最佳分段情况,发现就是6,于是就不需要继续。
只需要将代码稍微修改即可达到目的:
#include<iostream> using namespace std; /* * 存储结果的结构体,里面包含r和s两个数组,分别保存最佳收益和最佳收益时的分段数值 */ struct result { int* r; int* s; int len; result(int l) : r(), s(), len(l) { r = new int[len]; s = new int[len]; r[0] = 0; } ~result() { delete[] r; delete[] s; } }; result* extended_bottom_up_cut_rod(int p[], int n) { result* res = new result(n + 1); int q = -1; //外层的循环代表的是保留的不切割的那段 for (int i = 1; i <= n; i++) { //内层的循环代表的是要分割的,且要求出最佳分割的那段 for (int j = 1; j <= i; j++) { if (q < p[j] + res->r[i - j]) { q = p[j] + res->r[i - j]; res->s[i] = j; } } res->r[i] = q; } return res; } int main() { int p[] = { 0, 1, 5, 8, 9, 10, 17, 17, 20, 24, 30 }; int n = 9; result* res = extended_bottom_up_cut_rod(p, n); cout << "最佳收益:" << res->r[9] << endl; //循环输出实际的最佳分割段长 cout << "分段情况:"; while (n > 0) { cout << res->s[n] << ' '; n = n - res->s[n]; } delete res; return 0; }
运行上面程序,我们就可以的得到长度为9的钢管的最佳收益以及对应的切割情况:
最佳收益:25
分段情况:3 6
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。