混合背包问题
题目描述
有 N 种物品和一个容量是 V 的背包。物品一共有三类,每种体积是 vi,价值是 wi。
第一类物品只能用1次(01背包);
第二类物品可以用无限次(完全背包);
第三类物品最多只能用 si 次(多重背包);
求解将哪些物品装入背包,可使物品体积总和不超过背包容量,且价值总和最大。输出最大价值。
输入格式
第一行两个整数,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
分析:
原来的二维dp表每一行的物品都是同样类型的物品,可以用同样的处理手段,
现在每一行的物品可能是不一样的物品,因此对每一行进行更新的时候,先判断它的物品类型,再用相应的方法进行求解,更新该行。
主要可以分为两种: 完全背包和多重背包,01背包就是物品使用次数为1的多重背包。
分析不同情况,进行状态转移即可,如下图:
代码实现
#include <iostream>
#define read(x) scanf("%d",&x)
using namespace std;
const int maxn=1010,maxv=1010;
int v[maxn],w[maxn],s[maxn];
int dp[maxv];
int main() {
int N,V;
read(N),read(V);
for (int i=1;i<=N;i++) read(v[i]),read(w[i]),read(s[i]);
for (int i=1;i<=N;i++) {
if (s[i]==0) { //完全背包
for (int j=v[i];j<=V;j++) dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
} else {
if (s[i]==-1) s[i]=1; //0-1背包就是特殊的多重背包
for (int k=1;k<=s[i];k*=2) {//每个背包挨个拆分
for (int j=V;j>=k*v[i];j--) dp[j]=max(dp[j],dp[j-k*v[i]]+k*w[i]);
s[i]-=k;
}
if(s[i]) {
for (int j=V;j>=s[i]*v[i];j--) dp[j]=max(dp[j],dp[j-s[i]*v[i]]+s[i]*w[i]);
}
}
}
printf("%d",dp[V]);
return 0;
}
也可以节省内存,读入一次,处理一次,一次处理一行。
#include <iostream>
using namespace std;
const int N = 1010;
int n, m;
int f[N];
int main() {
cin >> n >> m;
for (int i = 0; i < n; i++) {
int v, w, s;
cin >> v >> w >> s;
if (s == 0) { // 完全背包
for (int j = v; j <= m; j++) f[j] = max(f[j], f[j - v] + w);
} else {
if (s == -1) s = 1; // 全部转为多重背包,然后使用二进制优化
for (int k = 1; k <= s; k *= 2) {
for (int j = m; j >= k * v; j--)
f[j] = max(f[j], f[j - k * v] + k * w);
s -= k;
}
if (s) {
for (int j = m; j >= s * v; j--)
f[j] = max(f[j], f[j - s * v] + s * w);
}
}
}
cout << f[m] << endl;
return 0;
}