4.混合背包问题
问题:
有N件物品和一个容量是V的背包。
物品一共有三类:
第一类:物品只能用1次(01背包)
第二类:物品可以用无限多次(完全背包)
第三类:物品最多只能用si次(多重背包)
每种的体积是vi,价值是wi。
求解将哪些物品装入背包,可使这些物品的总体积不超过背包的容量,且价值总和最大。
输入格式
第一行有两个整数,N,V用空格隔开,分别表示物品种数和背包容积。
接下来有N行,每行三个整数vi,wi,si,用空格隔开,分别表示第i种物品的体积、价值和数量。
si=-1:表示第i种物品只能用1次
si=0:表示第i种物品可以用无限次
si>0:表示第i种物品可以使用si次
输出格式
输出一个整数,表示最大价值
数据范围
0<N,V<=1000
0<vi,wi<=1000
-1<=si<=1000
输入样例
4 5
1 2 -1
2 4 1
3 4 0
4 5 2
输出样例
8
分析思路:
分成若干份
判断当前物品是哪一类的,转移相应的类
01背包:从大到小枚举
完全背包:从小到大枚举
代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int N=1010;
int n,m; //n个物品,m总容量
int f[N];
struct Good
{
int kind;
int v,w; //体积,价值
};
vector<Good>goods; //表示物品
int main(){
cin>>n>>m;
for(int i=0;i<n;i++) //枚举所有物品
{
int v,w,s; //体积,价值,个数
cin>>v>>w>>s;
if(s<0)goods.push_back({-1,v,w}); //01背包
else if(s==0) goods.push_back({-1,v,w}); //完全背包,输入信息
else
{ //多重背包问题
for(int k=1;k<=s;k*=2) //枚举分成哪些份
{
s-=k;
goods.push_back({-1,v*k,w*k}); //把k份物品直接放入 ,类别是01背包
}
if(s>0)goods.push_back({-1,v*s,w*s});
}
}
for(auto good:goods) //遍历所有物品
{
if(good.kind<0) //01背包
{
for(int j=m;j>=good.v;j--) //从大到小枚举
f[j]=max(f[j],f[j-good.v]+good.w);
}
else
{
for(int j=thing.v;j<=m;j++) //完全背包,从小到大枚举
f[j]=max(f[j],f[j-good.v]+good.w);
}
}
cout<<f[m]<<endl;
return 0;
}