动态规划解0-1背包问题

本博文为博主自己对0-1背包问题的理解并结合网上的博客所写,因为个人水平有限,若是有错误的地方欢迎指出。谢谢!

  •  问题描述:

给定n种物品和一个背包,物品i的重量是w[i],其价值为v[i],背包容量为cap,如何选择转入背包的物品,使得装入背包中的物品价值总最大?

  •  问题归纳

 对于某一种物品,要么装入背包,要么不装,我们将其状态取0-1,装入为1,不装为0,此问题称为0-1背包问题。

一、问题分析及代码实现

当w[ ]={2,2,6,5,4};v[ ]={6,3,5,4,6},背包容量cap=10时

1、新建一个二维数组dp[6][11],dp[i][j]为背包容量为j,第i个物品装入背包时的最大价值。

为什么要多一行一列了?因为这时要考虑,不往背包装物品和背包容量为0时的情况,此时。装入第一个物品时,只有当cap>=w[1],如下图:

2、上图中,只有一个物品w[1 ],所以就算cap容量再大也只放一个,当装物品2时,只考虑1、2不考虑剩下的,只考虑已经装的和即将装入的价值。同理得到:

其中,蓝色的框表示,当考虑物品 i 时,虽然,此时背包容量等于w[ i ],但是其价值没有之前的同等容量下的价值大,所以维持不变。绿色表示,当此时背包的容量减去当前物品的重量以后,还有多余的容量,而多余的容量能装下的物品大的价值大与之前的方案,所以修改装入方案,使价值最大。

3、递推式:

当当cap<w[i],说明装不下当前的,那此时价值应该和之前的相同;当cap>=w[i],要考虑装入目前的了,多余的空间装入的价值和之前方案的价值的比较,若小于,则当前物品不用装入,若大于就装入;

1 dp=  dp[i-1][j];                                //当cap<w[i]
2      max(dp[i-1][j],dp[i-1][j-w[i]]+v[i]);      //当cap>=w[i]

 4、代码实现:

 1 /*说明:
 2  *w[]为每个物体的重量,v[]为每个物体的价值,cap为背包的容量
 3  */
 4 int Knapsack(const vector<int> &w,const vector<int> &v,const int cap)
 5 {
 6     int n=w.size();        //亦为物体的个数
 7     vector<vector<int>> dp(n+1,vector<int>(cap+1,0));
 8     int i,j;
 9     for(i=1;i<=n;i++)
10     {
11         for(j=1;j<=cap;j++)
12         {
13             dp[i][j]=dp[i-1][j];
14             if(j>=w[i-1])
15             {
16                 dp[i][j]=max(dp[i-1][j],dp[i-1][j-w[i]]+v[i-1]);
17             }
18         }
19     }
20     return dp[n+1][cap+1];
21 }

 这里也可以用数组而不用向量,但因数组作为形参的时候,不能引入数组大小,函数要增加一个形参,所以图省事就用了向量。另外,这里代码中w[i-1]和v[i-1]之所以要减1,是因为二维矩阵中的下标与重量w[ ]和价值v[ ]的下标相差1要记得转换。

二、空间优化

 注意到一点,就是每当我们计算dp[i][j]时,只用到了dp[i-1][0...j],因此,我们可以只用一个一维数组来实现上面的功能。

 这里w[ ]={2,2,6,5,4};其下标不用转换。

 1 int Knapsack(const vector<int> &w,const vector<int> &v,const int cap)
 2 {
 3     int n=w.size();        //亦为物体的个数
 4     vector<int> dp(cap+1,0);
 5     int i;
 6     for(i=0;i<n;i++)    
 7     {
 8         for(j=cap;j>=0;j--)
 9         {
10             if(w[i]<j)
11             {
12                 dp[j]=max(dp[j],dp[j-w[i]]+v[i]);
13             }
14         }
15     }
16     return dp[cap+1];
17 }

 值得注意的是:

dp[ ]是从右往左更新的,因为若是从左往右更新,其后的的值会受到之前已更新的值的影响,达不到不优化空间之间的访问上一行中元素的效果。这一点在很多优化动态规划算法的空间时都要注意。

 

Ref:

http://blog.csdn.net/kangroger/article/details/38864689 

http://blog.csdn.net/dapengbusi/article/details/7463968

http://blog.csdn.net/stack_queue/article/details/53544109(背包九讲)

posted @ 2017-08-16 17:16  王大咩的图书馆  阅读(417)  评论(0编辑  收藏  举报