01背包:
有N件物品和一个容量为M的背包。放入第i件物品耗费的空间是W[i],得到的价值是V[i]。求解将哪些物品装入背包可使价值总和最大。
解题思路:每种物品仅有一件,可以选择放或不放。
则其状态转移方程便是: DP[i,j]=max{DP[i−1,j],DP[i−1,j−Wi]+Vi}
在这里,我用了c++的动态创建二维数组,如果用java动态创建数组就简单很多了。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
int main()
{
cout << "请输入物品的种类和背包的容量:" << endl;
int n,m;
cin>>n>>m;
cout<<"分别输入N种商品的重量和价值:"<<endl;
int *w=new int [n+1];
int *v=new int [n+1];
for(int i=1;i<=n;i++)
scanf("%d %d",&w[i],&v[i]);
int **dp;
dp = new int*[n+1];
for(int i=0;i<=n;i++)
dp[i]=new int [m+1];
for(int i=0;i<=n;i++)
dp[i][0]=0;
for(int j=0;j<=m;j++)
dp[0][j]=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(j<w[i])
dp[i][j]=dp[i-1][j];
else
dp[i][j]=max(dp[i-1][j-w[i]]+v[i],dp[i-1][j]);
}
}
printf("该背包能装的物品的最大价值为:%d\n",dp[n][m]);
printf("放入的商品有:\n");
vector<int> res;
int t=m;
for(int i=n;i>0;i--){
if(dp[i][t]>dp[i-1][t]){
res.push_back(i);
t-=w[i];
}
}
for(int i=res.size()-1;i>=0;i--){
printf("商品%d: %d %d\n",res[i],w[res[i]],v[res[i]]);
}
return 0;
}
如果不需要输出放入的有哪些物品, 就可以对空间进行压缩,用一维数组就够了。
for(int i=0;i<n;i++)
for(int j=m;j>=w[i];j--)//这个遍历顺序很重要
dp[j]=max(dp[j],dp[j-w[i]]+v[i]);
如果将上面的第二重for循环的顺便反过来:
for(int i=0;i<n;i++)
for(int j=m;j>=w[i];j--)
dp[j]=max(dp[j],dp[j-w[i]]+v[i]);
这就变成了完全背包问题,即每个物品都有无限多个可以使用的。
至于为什么这样可以,我之前研究了下背包九讲上说的,不是很理解。
然后自己写了一个测试代码,输出了每一次for循环对dp的改变,
这样就很容易看懂了。
部分背包问题:
有N个商品,每个商品的重量为WI,价格为:VI,现有一个背包,最多能装M的重量.
问:怎样装能使包中装入的商品价值最高(对于每个商品可以只装该商品的一部分)
因为物品可以分割,贪心的思路就很容易想到了,只需要尽量往里面放单位价值最大的物品就好。
这里我用了结构体,主要复习了一下结构体的sort排序吧。
#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
typedef struct{
int index;
float w;
float v;
}product;
bool cmp(product p1,product p2){
return p1.v/p1.w>p2.v/p2.w;
}
int main()
{
product a[1000];
int n;
float m;
cout << "请输入物品的种类和背包的容量:" << endl;
scanf("%d %f",&n,&m);
cout<<"分别输入N种商品的重量和价值:"<<endl;
for(int i=0;i<n;i++){
scanf("%f %f",&a[i].w,&a[i].v);
a[i].index=i+1;
}
sort(a,a+n,cmp);
printf("先对商品进行单位价值递减进行排序:\n");
for(int i=0;i<n;i++)
printf("商品%d: %.2f %.2f\n",a[i].index,a[i].w,a[i].v);
float ans=0;
printf("装入的物品编号及放入的数量\n");
for(int i=0;i<n;i++){
if(m>=a[i].w){
printf("%d %.2f\n",a[i].index,a[i].w);
m-=a[i].w;
ans+=a[i].v;
}else{
printf("%d %.2f\n",a[i].index,m);
ans+=m/a[i].w*a[i].v;
break;
}
}
printf("能装入的商品的最大价值为:%.2f",ans);
return 0;
}