B-01背包问题
题目
时间限制: 1 Sec 内存限制: 64 MB
题目描述
有 n 件物品, 每件物品有一个价值和一个重量,分别记为: b1,b2, …bn w1,w2, …wn 其中所有的 重量wi 均为整数。 现有一个背包,其最大载重量为W,要求从这n件物品中任取若干件(这些物品要么被装入要么被留下)。问背包中装入哪些物品可使得所装物品的价值和最大?
输入
第1行:2个整数n(1<=n<=1000)和W(1<=W<=10000),分别表示物品的件数和背包的最大载重量。
第2-n+1行:每行2个用空格分开的整数,第i+1行的整数表示第i件物品的重量wi和价值bi(1<=bi,wi<=10000)。
输出
第1行:1个整数,表示背包所能装下的物品的最大总价值。
第2-?行:每行3个用空格分开的整数,i, wi, bi,分别表示最优解中的物品的编号、重量和价值。
样例输入
4 5
2 3
3 4
4 5
5 6
样例输出
7
1 2 3
2 3 4
状态定义
f[i][j]
:前
i
个物品,选出重量和恰好为
状态转移
分2种情况,选
i
物品和不选
- 选
i
:
f[i][j]=f[i−1][j−w[i]]+b[i] ; - 不选
i
:
f[i][j]=f[i−1][j] 。
但我们不可能两种情况都要,所以取大的一个:
f[i][j]=max(f[i−1][j−w[i]]+b[i],f[i−1][j])
。
边界
无。
输出解
用 Ans[i][j] 表示 f[i][j] 是由哪个状态转移得来,即:
- f[i][j]==f[i−1][j−w[i]]=>Ans[i][j]=1
- f[i][j]==f[i−1][j]=>Ans[i][j]=0
然后递归输出即可。
代码
#include<cstdio>
#include<algorithm>
using namespace std;
int read()
{
int x=0,f=1;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
return x*f;
}
#define MAXN 1000
#define MAXW 10000
int w[MAXN+5],b[MAXN+5];
int f[MAXN+5][MAXW+1];
bool Ans[MAXN+5][MAXW+5];
int N,W;
void print(int i,int j)
{
if(i<1||j<0) return;
if(Ans[i][j])//当Ans[i][j]==1时,f[i][j]=f[i-1][j-w[i]]+b[i],表示要选i
{
print(i-1,j-w[i]);
printf("%d %d %d\n",i,w[i],b[i]);
}
else print(i-1,j);//否则是不选i的,就不用输出
}
int main()
{
N=read(),W=read();
for(int i=1;i<=N;i++)
w[i]=read(),b[i]=read();
for(int i=1;i<=N;i++)
for(int j=0;j<=W;j++)
{
if(j>=w[i]&&f[i-1][j]<f[i-1][j-w[i]]+b[i])
{
f[i][j]=f[i-1][j-w[i]]+b[i];
Ans[i][j]=1;//其实不用Ans也可以,在递归的时候比较f[i][j]和f[i-1][j-w[i]]+b[i]即可
}
else f[i][j]=f[i-1][j];
}
printf("%d\n",f[N][W]);
print(N,W);
}
优化-滚动数组
和A题一样,
f[j]
在更新前实际上就是
f[i−1][j]
的值,所以可以用一维完成。
但是这样就不能不用
Ans
数组了。
详细内容请看:为什么01背包要逆序循环。
#include<cstdio>
#include<algorithm>
using namespace std;
int read()
{
int x=0,f=1;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
return x*f;
}
#define MAXN 1000
#define MAXW 10000
int w[MAXN+5],b[MAXN+5];
int f[MAXW+1];
bool Ans[MAXN+5][MAXW+5];
int N,W;
void print(int i,int j)
{
if(i<1||j<0) return;
if(Ans[i][j])
{
print(i-1,j-w[i]);
printf("%d %d %d\n",i,w[i],b[i]);
}
else print(i-1,j);
}
int main()
{
N=read(),W=read();
for(int i=1;i<=N;i++)
w[i]=read(),b[i]=read();
for(int i=1;i<=N;i++)
for(int j=W;j>=0;j--)//倒着来
if(j>=w[i]&&f[j]<f[j-w[i]]+b[i])
{
f[j]=f[j-w[i]]+b[i];
Ans[i][j]=1;
}
printf("%d\n",f[W]);
print(N,W);
}