编程算法 - 完全背包问题 代码(C)

完全背包问题 代码(C)

本文地址: http://blog.csdn.net/caroline_wendy

题目: 有n个重量和价值分别为w,v的物品, 从这些物品中挑选出总重量不超过W的物品, 求所有挑选方案中价值总和的最大值. 

*每件物品可以挑选任意多件.

动态规划: 每次选取最大的组合, 加入到数组, 第一种时间复杂度O(nW^2), 第二种时间复杂度O(nW).

解法1, max()部分表明, 要么来源于上面, 要么来源于前面.

代码:

[cpp]  view plain copy

1. /* 
2.  * main.cpp 
3.  * 
4.  *  Created on: 2014.7.17 
5.  *      Author: spike 
6.  */  
7.   
8. /*eclipse cdt, gcc 4.8.1*/  
9.   
10. #include <stdio.h>  
11. #include <memory.h>  
12. #include <limits.h>  
13.   
14. #include <utility>  
15. #include <queue>  
16. #include <algorithm>  
17.   
18. using namespace std;  
19.   
20. class Program {  
21.     static const int MAX_N = 100;  
22.   
23.     int n=4, W=5;  
24.     int w[MAX_N] = {2,1,3,2}, v[MAX_N]={3,2,4,2};  
25.     int dp[MAX_N+1][MAX_N+1];  
26. public:  
27.     void solve() {  
28.         for (int i=0; i<n; ++i) {  
29.             for (int j=0; j<=W; j++) {  
30.                 for (int k=0; k*w[i] <= j; k++) {  
31.                     dp[i+1][j] = max(dp[i+1][j], dp[i][j-k*w[i]]+k*v[i]);  
32.                 }  
33.             }  
34.         }  
35.         printf("result = %d\n", dp[n][W]);  
36.     }  
37.   
38.     void solve1() {  
39.         for (int i=0; i<n; ++i) {  
40.             for (int j=0; j<=W; j++) {  
41.                 if (j<w[i]) {  
42.                     dp[i+1][j] = dp[i][j];  
43.                 } else {  
44.                     dp[i+1][j] = max(dp[i][j], dp[i+1][j-w[i]]+v[i]);  
45.                 }  
46.             }  
47.         }  
48.         printf("result = %d\n", dp[n][W]);  
49.     }  
50. };  
51.   
52.   
53. int main(void)  
54. {  
55.     Program P;  
56.     P.solve1();  
57.     return 0;  
58. }  

输出:

[plain]  view plain copy

1. result = 10  

为了节约内存, 可以使用一维数组进行求解.

代码:

[cpp]  view plain copy

1. /* 
2.  * main.cpp 
3.  * 
4.  *  Created on: 2014.7.17 
5.  *      Author: spike 
6.  */  
7.   
8. /*eclipse cdt, gcc 4.8.1*/  
9.   
10. #include <stdio.h>  
11. #include <memory.h>  
12. #include <limits.h>  
13.   
14. #include <utility>  
15. #include <queue>  
16. #include <algorithm>  
17.   
18. using namespace std;  
19.   
20. class Program {  
21.     static const int MAX_N = 100;  
22.   
23.     int n=3, W=7;  
24.     int w[MAX_N] = {3,4,2}, v[MAX_N]={4,5,3};  
25.     int dp[MAX_N+1];  
26. public:  
27.     void solve() {  
28.         memset(dp, 0, sizeof(dp));  
29.         for (int i=0; i<n; ++i) {  
30.             for (int j=w[i]; j<=W; j++) {  
31.                 dp[j] = max(dp[j], dp[j-w[i]]+v[i]);  
32.             }  
33.         }  
34.         printf("result = %d\n", dp[W]);  
35.     }  
36. };  
37.   
38.   
39. int main(void)  
40. {  
41.     Program P;  
42.     P.solve();  
43.     return 0;  
44. }  

输出:

[plain]  view plain copy

1. result = 10  

可以讲两个数组滚动使用, 节省内存.

代码:

[cpp]  view plain copy

1. /* 
2.  * main.cpp 
3.  * 
4.  *  Created on: 2014.7.17 
5.  *      Author: spike 
6.  */  
7.   
8. /*eclipse cdt, gcc 4.8.1*/  
9.   
10. #include <stdio.h>  
11. #include <memory.h>  
12. #include <limits.h>  
13.   
14. #include <utility>  
15. #include <queue>  
16. #include <algorithm>  
17.   
18. using namespace std;  
19.   
20. class Program {  
21.     static const int MAX_N = 100;  
22.   
23.     int n=3, W=7;  
24.     int w[MAX_N] = {3,4,2}, v[MAX_N]={4,5,3};  
25.     int dp[2][MAX_N+1];  
26. public:  
27.     void solve() {  
28.         memset(dp,0,sizeof(dp));  
29.         for (int i=0; i<n; ++i) {  
30.             for (int j=0; j<=W; j++) {  
31.                 if (j<w[i]) {  
32.                     dp[(i+1)&1][j] = dp[i&1][j];  
33.                 } else {  
34.                     dp[(i+1)&1][j] = max(dp[i&1][j], dp[(i+1)&1][j-w[i]]+v[i]);  
35.                 }  
36.             }  
37.         }  
38.         printf("result = %d\n", dp[n&1][W]);  
39.     }  
40. };  
41.   
42.   
43. int main(void)  
44. {  
45.     Program P;  
46.     P.solve();  
47.     return 0;  
48. }  

输出:

[plain]  view plain copy

1. result = 10  

当重量可选范围很大, 价值可选范围小的时候, 可以使根据价值进行动态规划.

代码:

[cpp]  view plain copy

1. /* 
2.  * main.cpp 
3.  * 
4.  *  Created on: 2014.7.17 
5.  *      Author: spike 
6.  */  
7.   
8. /*eclipse cdt, gcc 4.8.1*/  
9.   
10. #include <stdio.h>  
11. #include <memory.h>  
12. #include <limits.h>  
13.   
14. #include <utility>  
15. #include <queue>  
16. #include <algorithm>  
17.   
18. using namespace std;  
19.   
20. class Program {  
21.     static const int MAX_N = 100;  
22.     static const int MAX_V = 4;  
23.     const int INF = INT_MAX>>2;  
24.   
25.     int n=4, W=5;  
26.     int w[MAX_N] = {2,1,3,2}, v[MAX_N] = {3,2,4,2};  
27.   
28.     int dp[MAX_N+1][MAX_N*MAX_V+1];  
29. public:  
30.     void solve() {  
31.         fill(dp[0], dp[0]+MAX_N*MAX_V+1, INF);  
32.   
33.         dp[0][0] = 0;  
34.         for (int i=0; i<n; ++i) {  
35.             for (int j=0; j<=MAX_N*MAX_V; j++) {  
36.                 if (j<v[i]) {  
37.                     dp[i+1][j] = dp[i][j];  
38.                 } else {  
39.                     dp[i+1][j] = min(dp[i][j], dp[i][j-v[i]]+w[i]);  
40.                 }  
41.             }  
42.         }  
43.         int res = 0;  
44.         for (int i=0; i<=MAX_N*MAX_V; ++i) if (dp[n][i]<=W) res = i;  
45.         printf("result = %d\n", res);  
46.     }  
47. };  
48.   
49.   
50. int main(void)  
51. {  
52.     Program P;  
53.     P.solve();  
54.     return 0;  
55. }  

输出:

[plain]  view plain copy

1. result = 7