您的位置:首页 > 其它

HDU 2602 Bone Collector (01背包问题 基础)

2015-10-07 16:23 211 查看

题目链接:HDU 2602 Bone Collector

题目描述:


输入 T 代表 T组数据, 每组数据一共3行,第一行输入n 和 m ,分别代表可选择物品的数量和背包的大小

接下来两行每行 n 个数字, 第一行数字代表每个物品的价值,第二行代表每个物品的体积。

问,给定背包能装物品的最大价值是多少。



Sample Input

1
5 10
1 2 3 4 5
5 4 3 2 1

Sample 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
[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]);
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: