算法设计与分析之不定期更新的日常 0-1背包问题
给定n种物品和一个背包。物品i的重量是wi,其价值为vi,背包的容量为c。问应该如何选择装入背包中的物品,使得装入背包中的物品的总价值最大?
(整数背包的问题)
代码实现:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=100;
void Knapsack(int v[],int w[],int c,int n,int m[][maxn])
{
int jMax=min(w[n]-1,c);
for(int j=0;j<=jMax;j++)
{
m[n][j]=0;
}
for(int j=w[n];j<=c;j++)
{
m[n][j]=v[n];
}
for(int i=n-1;i>1;i--)
{
jMax=min(w[i]-1,c);
for(int j=0;j<=jMax;j++)
{
m[i][j]=m[i+1][j];
}
for(int j=w[i];j<=c;j++)
{
m[i][j]=max(m[i+1][j],m[i+1][j-w[i]]+v[i]);
}
}
m[1][c]=m[2][c];
if(c>=w[1])
m[1][c]=max(m[1][c],m[2][c-w[1]]+v[1]);
}
void Traceback(int m[][maxn],int w[],int c,int n,int x[])
{
for(int i=1;i<n;i++)
{
if(m[i][c]==m[i+1][c])
x[i]=0;
else
{
x[1]=1;
c-=w[i];
}
}
x[n]=(m[n][c])?1:0;
}
int main()
{
int n,c;
int w[maxn],v[maxn];
int m[maxn][maxn];///二维数组用来存储m(i,j)的相应的值,m(i,j)表示背包容量为j,
///可选择的物品为从第i个到第n个
int x[maxn];///用于存放构造出的最优解
while(scanf("%d%d",&n,&c)!=EOF)
{
memset(w,0,sizeof(w));
memset(v,0,sizeof(v));
memset(m,0,sizeof(m));
memset(x,0,sizeof(x));
for(int i=1;i<=n;i++)
{
scanf("%d",&w[i]);
}
for(int i=1;i<=n;i++)
{
scanf("%d",&v[i]);
}
Knapsack(v,w,c,n,m);
Traceback(m,w,c,n,x);
for(int i=1;i<=n;i++)
{
if(x[i])
printf("%d ",i);
}printf("\n");
}
return 0;
}
/*
5 10
2 2 6 5 4
6 3 5 4 6
*/