AcWing 7. 混合背包问题

AcWing 7. 混合背包问题

有 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背包，完全背包和多重背包的结合。

对于01背包和完全背包，区别就在于前者将体积从大到小枚举，后者将体积从小到大枚举。

多重背包可以将相同的物品拆成单个物品变成01背包。但这题数据量很大，可以用二进制优化的方法来拆分相同的物品，将s个相同的物品拆成若干个2^n物品合并的形式，最后s - 2^n 个物品合并成一个物品，这样对于1 ~ s数量的物品都可以用我们拆分的物品表示出来，对于拆出来的物品我们可以用01背包的问题解决。
多重背包

所以最后对于标记是01背包的物品我们将体积从大到小枚举，完全背包的物品将体积从小到大枚举我们就可以得出答案了。

AC代码:

#include <iostream>
#include <vector> 
using namespace std;
const int N = 1e6 + 5;
struct Thing { // 物品 
	int v, w, s;
};
int f[N];
int main() {
	int n, m;
	cin >> n >> m;
	vector<Thing> things;
	for (int i = 1; i <= n; i ++) {
		int v, w, s;
		cin >> v >> w >> s;
		if (s == -1) things.push_back({v, w, s}); // 01背包物品 
		else if(s == 0) things.push_back({v, w, s}); // 完全背包物品 
		else {
			for (int j = 1; j <= s; j *= 2) { // 多重背包的物品拆成2^n形式的物品合并 
				s -= j;
				things.push_back({j * v, j * w, -1}); // 标记为01背包物品 
			}
			if (s > 0) things.push_back({s * v, s * w, -1}); // 最后s - 2^n个物品合并 
		}
	}
	for (auto thing : things) {
		if (thing.s == -1) { // 01背包物品体积从大到小进行枚举 
			for (int j = m; j >= thing.v; j --) 
				f[j] = max(f[j], f[j - thing.v] + thing.w);
		}
		else { // 完全背包物品体积从小到大进行枚举 
			for (int j = thing.v; j <= m; j ++) 
				f[j] = max(f[j], f[j - thing.v] + thing.w);
		}
	}
	cout << f[m] << '\n';
	return 0;
}