输入 T 代表 T组数据, 每组数据一共3行,第一行输入n 和 m ,分别代表可选择物品的数量和背包的大小
接下来两行每行 n 个数字, 第一行数字代表每个物品的价值,第二行代表每个物品的体积。
问,给定背包能装物品的最大价值是多少。
Sample Input
1 5 10 1 2 3 4 5 5 4 3 2 1Sample Output
14
如样例 , 选择第 2 ,3 ,4 ,5 个物品正好装满背包,总价值为 14,为最大的情况。
我们用 value[ i ]储存每个物品的价值,用vol[ i ]储存每个物品的体积 (i >= 1) ,
引入dp[ i ][ j ], 代表把前 i 个物品装到容量为 j 的背包中的最大价值。
01背包的特点就是:每种物品仅有一件,可以选择放或不放。
所以状态转移方程为 dp[ i ][ j ] = max( dp[ i - 1 ][ j ] , dp[ i - 1 ][ j - vol[ i ] + value[ i ]);
这个方程其实不难理解 , 就是当前背包中最大价值 等于 (不放第 i 件物品 和 放第 i 件物品 这两者的最大值)
这里可能对 dp[ i - 1 ][ j - vol[ i ] + value[ i ] 有疑惑, i - 1 代表上一层的状态, 由于上一层我们 将 容量 0 ~m的状态全部计算了出来,所以可以利用
dp[ i - 1 ][ j - vol[ i ] ] 就代表了上一层状态中 背包容量为 j - vol[ i ] 时所能容纳的最大价值,然后在加上 value[ i ], 也就是当前物品的最大价值。
这样我们从第一件物品的放或不放, 一直计算到第n件物品,所以结果就是 dp[ n ][ m ].
这里有一个对空间的优化版本,我们还可以把dp数组变成 1 维的。
直接给出代码
memset(DP,0,sizeof(DP)); for(int i=1;i<=n;i++) for(int j=m;j>=0;j--){ if(j>=vol[i]) DP[j] = max(DP[j],DP[j-vol[i]] + value[i]); } printf("%d\n",DP[m]);
为什么这么做可以呢? 注意看for循环的计算顺序,DP数组是从上到下,从右向左计算的。计算dp[ i ][ j ]之前,DP[ j ]中保存的就是 dp[ i - 1 ][ j ] ,
而 DP[ j - vol [i] ] ,储存的就是dp[ i - 1 ][ j - w ] ,而不是 dp [ i ][ j - w ], 因为 我们的 j 是逆序枚举的, 此时 dp [ i ][ j - w ]还没有算出来 。
这样 , 用一维数组就能储存上一层的状态了。
#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> using namespace std; const int maxn = 1010; int value[maxn]; int vol[maxn]; int dp[maxn][maxn]; int DP[maxn]; int main(){ int t; scanf("%d",&t); while(t--){ int n,m; scanf("%d%d",&n,&m); memset(dp,0,sizeof(dp)); for(int i=1;i<=n;i++) scanf("%d",&value[i]); for(int i=1;i<=n;i++) scanf("%d",&vol[i]); for(int i=1;i<=n;i++){ for(int j=0;j<=m;j++){ //vol容量可能为0; j 的顺序无所谓 if(j>=vol[i]) dp[i][j] = max(dp[i-1][j] , dp[i-1][j-vol[i]] + value[i]); else dp[i][j] = dp[i-1][j]; } } // for(int i=0;i<=n;i++){ //可以打印出来看看手动模拟下 // for(int j=0;j<=m;j++) // printf("%d ",dp[i][j]); // printf("\n"); // } printf("%d\n",dp[n][m]); // memset(DP,0,sizeof(DP)); //空间优化方法 // for(int i=1;i<=n;i++) // for(int j=m;j>=0;j--){ j的顺序必须从后往前 // if(j>=vol[i]) // DP[j] = max(DP[j],DP[j-vol[i]] + value[i]); // } // printf("%d\n",DP[m]); } return 0; }
第一种方法还有一种与之“对称”的状态定义 用 dp [ i ][ j ],表示把 i , i + 1 , i + 2 , , , n 装入容量为 j 的 背包中的最大价值
显然 外层循环需要从 n ~1,而最后的答案为 dp [ 1 ][ m ]
memset(dp,0,sizeof(dp)); for(int i=n;i>=1;i--){ for(int j=0;j<=m;j++){ if(j>=vol[i]) dp[i][j] = max(dp[i+1][j] , dp[i+1][j-vol[i]] + value[i]); else dp[i][j] = dp[i+1][j]; } } printf("%d\n",dp[1][m]);