【算法导论】动态规划之“钢管切割”问题

地址：http://blog.csdn.net/cyp331203/article/details/42806159

动态规划算法导论递归重构钢管切割

目录(?)[+]

        动态规划，其实跟分治法有些相似，基本思想都是将复杂的问题分成数个简单的子问题，然后再去解决。它们的区别在于，分治法关注的子问题不相互“重叠”，而动态规划关注的子问题，多是相互“重叠”的。比如在快速排序中，我们将数据分成两部分，这两部分再分别快速排序的递归思想，也就是将整个问题的排序划分为子问题子数组的排序，但是这两个子数组的排序之间并没有相互联系，a子数组的排序不会因为b子数组的排序而得到任何“好处”或者“坏处”。但是有些时候，划分的子问题之间却是有联系的，比如下面的“钢管切割”问题：

        钢管切割原始问题：

        某公司生产长钢管，然后一般，会将钢条切断，变成不同长度，然后去售卖。其中有个问题是，不同长度的钢管的售价是不一样的，但是它们并不是完全按照比例来，比如2米的钢管售价要比3米的钢管售价要少，但是并不是2比3的比例。钢管的长度售价表如下：

 长度i1      2      3      4      5      6      7      8      9      10价格Pi1      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。

        于是给以下实现代码：

[cpp] view plaincopy

1. int cut_rod(int* p, int n) {  
2.     if (n == 0) {  
3.         return 0;  
4.     }  
5.     int q = -1;  
6.     for (int i = 1; i <= n; i++) {  
7.         /* 
8.          * 将n长的钢条，分成i和n-i的两段，i长的那段不切割，而n-i的那段求最大 
9.          * 切割收益方式，然后相加；而q值是所有的组合中，最大收益的那个 
10.          */  
11.         q = max(q, p[i] + cut_rod(p, n - i));  
12.     }  
13.     return q;  
14. }  

        这种方法比较容易理解，但是性能是不是好呢？

        可以简单的以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时的结果，只需要在第一次计算的时候将结果保存起来，然后再需要的时候直接使用即可。这其实就是所谓的动态规划算法。

        这里的思路有两种，一种叫带备忘的自顶向下方法，是顺着之前的代码，当需要的时候去检查是不是已经计算好了，如果是，则直接使用，如果不是，则计算，并保存结果。第二种思路是自底向上方法，不论需不需要，先将子问题一一解决，然后再来解决更一级的问题，但要注意的是，我们需要先从最小的子问题开始，依次增加规模，这样每一次解决问题的时候，它的子问题都已经计算好了，直接使用即可。

        带备忘的自顶向下方法:

[cpp] view plaincopy

1. int memoized_cut_rod_aux(int* p, int n, int* r) {  
2.     if (r[n] >= 0) {  
3.         return r[n];  
4.     }  
5.     int q = -1;  
6.     if (n == 0) {  
7.         q = 0;  
8.     } else {  
9.         for (int i = 1; i <= n; i++) {  
10.             q = max(q, p[i] + memoized_cut_rod_aux(p, n - i, r));  
11.         }  
12.     }  
13.     r[n] = q;  
14.     return q;  
15. }  
16.   
17. /* 
18.  * 自顶向上的cut-rod的过程 
19.  */  
20. int memoized_cut_rod(int* p, int n) {  
21.     int* r = new int[n + 1];  
22.   
23.     //初始化r数组，r数组用来存放，某种解决方案的最大收益值，对于n长的钢条而言，有n+1种切割方案，所以数组n+1长  
24.     for (int i = 0; i <= n; i++) {  
25.         r[i] = -1;  
26.     }  
27.     return memoized_cut_rod_aux(p, n, r);  
28. }  

        自底向上的方法：

[cpp] view plaincopy

1. /* 
2.  * 自底向上的方式，先计算更小的子问题，然后再算较大的子问题，由于较大的子问题依赖于更小的子问题的答案，所以在计算较 
3.  * 大的子问题的时候，就无需再去计算更小的子问题，因为那答案已经计算好，且存储起来了 
4.  */  
5.   
6. int bottom_up_cut_rod(int p[], int n) {  
7.   
8.     int* r = new int[n + 1];  
9.   
10.     r[0] = 0; //将r[0]初始化为0，是因为0长的钢条没有收益  
11.     for (int j = 1; j <= n; j++) {  
12.         int q = -1;  
13.   
14.         /* 
15.          * 这里不用i=0开始，因为i=0开始不合适，因为这里总长就是为j，而划分是i和j-i的划分，如果i等于0，那么 
16.          * 就意味着要知道r[j-0]=r[j]的值也就是j长的最好划分的收益，但是我们这里不知道。而且对于p[0]而言本身就没有意义 
17.          * p数组中有意义的数据下标是从1到n的 
18.          */  
19.         for (int i = 1; i <= j; i++) {  
20.             q = max(q, p[i] + r[j - i]); //  
21.         }  
22.         r[j] = q;  
23.     }  
24.     return r[n];  
25. }  

上面两种算法的时间复杂度都是O(n^2)。

        重构解

        上面的代码只给出了最优的收益值，但是却没有给出最优收益到底是在那种切割分配方式下得到的，比如说n=9时，最佳收益为25，要分成3和6两段。这里可以使用另一个数组s来存储分段情况，比如s[9]存储3，然后我们让n=9-3，就可以得到s[6]的最佳分段情况，发现就是6，于是就不需要继续。

只需要将代码稍微修改即可达到目的：

[cpp] view plaincopy

1. #include<iostream>  
2.   
3. using namespace std;  
4.   
5. /* 
6.  * 存储结果的结构体，里面包含r和s两个数组，分别保存最佳收益和最佳收益时的分段数值 
7.  */  
8. struct result {  
9.     int* r;  
10.     int* s;  
11.     int len;  
12.     result(int l) :  
13.             r(), s(), len(l) {  
14.         r = new int[len];  
15.         s = new int[len];  
16.         r[0] = 0;  
17.     }  
18.   
19.     ~result() {  
20.         delete[] r;  
21.         delete[] s;  
22.     }  
23. };  
24.   
25. result* extended_bottom_up_cut_rod(int p[], int n) {  
26.     result* res = new result(n + 1);  
27.     int q = -1;  
28.   
29.     //外层的循环代表的是保留的不切割的那段  
30.     for (int i = 1; i <= n; i++) {  
31.   
32.         //内层的循环代表的是要分割的，且要求出最佳分割的那段  
33.         for (int j = 1; j <= i; j++) {  
34.             if (q < p[j] + res->r[i - j]) {  
35.                 q = p[j] + res->r[i - j];  
36.                 res->s[i] = j;  
37.             }  
38.         }  
39.         res->r[i] = q;  
40.     }  
41.     return res;  
42. }  
43.   
44. int main() {  
45.     int p[] = { 0, 1, 5, 8, 9, 10, 17, 17, 20, 24, 30 };  
46.   
47.     int n = 9;  
48.   
49.     result* res = extended_bottom_up_cut_rod(p, n);  
50.   
51.     cout << "最佳收益：" << res->r[9] << endl;  
52.   
53.     //循环输出实际的最佳分割段长  
54.     cout << "分段情况：";  
55.     while (n > 0) {  
56.         cout << res->s[n] << ' ';  
57.         n = n - res->s[n];  
58.     }  
59.   
60.     delete res;  
61.   
62.     return 0;  
63. }  

运行上面程序，我们就可以的得到长度为9的钢管的最佳收益以及对应的切割情况：

最佳收益：25
分段情况：3 6