经典问题描述
有 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
测试题目链接 (AcWing)
混合背包问题就是01背包+完全背包+多重背包问题,根据物品的数量确定用哪种背包问题进行求解
下面给出两种解法组合
- 01背包滚动数组法+完全背包滚动数组法+多重背包二进制优化法
#include <iostream>
using namespace std;
#define Max(a, b) ((a > b) ? (a) : (b))
const int ArrMax{20010};
int dp[ArrMax], que[ArrMax], book[ArrMax];
int main(int argc, char **argv)
{
int N, V;
cin >> N >> V;
while (N--)
{
int v, w, s;
cin >> v >> w >> s;
if (s == -1)
for (int ind = V; ind >= v; --ind)
dp[ind] = Max(dp[ind], dp[ind - v] + w);
else if (!s || v * s >= V)
for (int ind = v; ind <= V; ++ind)
dp[ind] = Max(dp[ind], dp[ind - v] + w);
else
{
int n = 1;
while (1)
{
if (n < s)
{
for (int ind = V; ind >= n * v; --ind)
dp[ind] = Max(dp[ind], dp[ind - n * v] + n * w);
s -= n;
n <<= 1;
}
else
{
n = s;
for (int ind = V; ind >= n * v; --ind)
dp[ind] = Max(dp[ind], dp[ind - n * v] + n * w);
break;
}
}
}
}
cout << dp[V] << endl;
return EXIT_SUCCESS;
}
- 01背包滚动数组法+完全背包滚动数组法+多重背包单调队列优化法
#include <iostream>
using namespace std;
#define Max(a, b) ((a > b) ? (a) : (b))
const int ArrMax{20010};
int dp[ArrMax], que[ArrMax], book[ArrMax];
int main(int argc, char **argv)
{
int N, V;
cin >> N >> V;
while (N--)
{
int v, w, s;
cin >> v >> w >> s;
if (s == -1)
for (int ind = V; ind >= v; --ind)
dp[ind] = Max(dp[ind], dp[ind - v] + w);
else if (!s || v * s >= V)
for (int ind = v; ind <= V; ++ind)
dp[ind] = Max(dp[ind], dp[ind - v] + w);
else
for (int ind = 0; ind < v; ++ind)
{
int head{0}, tail{0};
for (int jnd = 0; jnd <= (V - ind) / v; ++jnd)
{
auto temp{dp[ind + jnd * v] - jnd * w};
while (head < tail && que[tail - 1] <= temp)
--tail;
book[tail] = jnd;
que[tail++] = temp;
if (book[head] < jnd - s)
++head;
dp[ind + jnd * v] = que[head] + jnd * w;
}
}
}
cout << dp[V] << endl;
return EXIT_SUCCESS;
}