题目地址:
https://www.acwing.com/problem/content/7/
有 N N N种物品和一个容量是 V V V的背包。物品一共有三类:第一类物品只能用 1 1 1次( 0 − 1 0-1 0−1背包);第二类物品可以用无限次(完全背包);第三类物品最多只能用 s i s_i si次(多重背包);每种体积是 v i v_i vi,价值是 w i w_i wi。求解将哪些物品装入背包,可使物品体积总和不超过背包容量,且价值总和最大。输出最大价值。
输入格式:
第一行两个整数,
N
N
N,
V
V
V,用空格隔开,分别表示物品种数和背包容积。接下来有
N
N
N行,每行三个整数
v
i
,
w
i
,
s
i
v_i,w_i,s_i
vi,wi,si,用空格隔开,分别表示第
i
i
i种物品的体积、价值和数量。
s
i
=
−
1
s_i=−1
si=−1表示第
i
i
i种物品只能用
1
1
1次;
s
i
=
0
s_i=0
si=0表示第
i
i
i种物品可以用无限次;
s
i
>
0
s_i>0
si>0表示第
i
i
i种物品可以使用
s
i
s_i
si次;
输出格式:
输出一个整数,表示最大价值。
数据范围:
0
<
N
,
V
≤
1000
0<N,V≤1000
0<N,V≤1000
0
<
v
i
,
w
i
≤
1000
0<v_i,w_i≤1000
0<vi,wi≤1000
−
1
≤
s
i
≤
1000
−1≤si≤1000
−1≤si≤1000
思路还是动态规划,状态表示是
f
[
i
]
[
j
]
f[i][j]
f[i][j],表示前
i
i
i个物品里选并且体积不超过
j
j
j的最大价值,可以按照第
i
i
i个物品选还是不选分类。对于
s
i
=
−
1
s_i=-1
si=−1,可以用
0
−
1
0-1
0−1背包的方式递推;对于
s
i
=
0
s_i=0
si=0,可以用完全背包的方式递推;对于
s
i
>
0
s_i>0
si>0,可以用多重背包的方式递推,但是这里需要加上二进制优化,以防超时。参考:
1、
0
−
1
0-1
0−1背包:https://blog.csdn.net/qq_46105170/article/details/113668186;
2、完全背包:https://blog.csdn.net/qq_46105170/article/details/113834289;
3、多重背包二进制优化版:https://blog.csdn.net/qq_46105170/article/details/113840962。
代码如下:
#include <iostream>
using namespace std;
const int N = 1010;
int n, m;
int f[N];
int main() {
scanf("%d%d", &n, &m);
for (int i = 0; i < n; i++) {
int v, w, s;
scanf("%d%d%d", &v, &w, &s);
if (!s)
for (int j = v; j <= m; j++) f[j] = max(f[j], f[j - v] + w);
else {
if (!~s) s = 1;
// 二进制优化,每次对k个当前物品打包做01背包
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);
}
}
printf("%d\n", f[m]);
}
时间复杂度 O ( V ( ( N − k ) + ∑ i = 1 k log s i ) ) O(V((N-k)+\sum _{i=1}^k\log s_i)) O(V((N−k)+∑i=1klogsi)),其中 k k k是可以使用 s i s_i si次的物品种类数(即要二进制优化的物品种类数),空间 O ( V ) O(V) O(V)。