背包九讲学习笔记
本文内容基本涵盖了 dd_engi 的背包九讲,在此基础上加上了自己的理解和代码实现
背包九讲原版 pdf 已由作者崔添翼(dd_engi)上传至 github,你可以 点击这里 进行下载
1. 01 背包问题
题目
有 N N N 件物品和一个容量为 V V V 的背包。放入第 i i i 件物品耗费的费用是 C i C_i Ci ,得到的价值是 W i W_i Wi 。求解将哪些物品装入背包可使价值总和最大。
基本思路
这是最基础的背包问题,特点是:每种物品仅有一件,可以选择放或不放。
用子问题定义状态:即
F
[
i
,
v
]
F[i,v]
F[i,v] 表示前
i
i
i 件物品恰放入一个容量为
v
v
v 的背包可以获得
的最大价值。则其状态转移方程便是:
F [ i , v ] = m a x { F [ i − 1 , v ] , F [ i − 1 , v − C i ] + W i } F[i, v] = max\{F[i - 1, v], F[i - 1, v - C_i] + W_i\} F[i,v]=max{F[i−1,v],F[i−1,v−Ci]+Wi}
这个方程非常重要,基本上所有跟背包相关的问题的方程都是由它衍生出来的。所以有必要将它详细解释一下:“将前 i i i 件物品放入容量为 v v v 的背包中”这个子问题,若只考虑第 i i i 件物品的策略(放或不放),那么就可以转化为一个只和前 i − 1 i − 1 i−1 件物品相关的问题。如果不放第 i i i 件物品,那么问题就转化为“前 i − 1 i − 1 i−1 件物品放入容量为 v v v 的背包中”,价值为 F [ i − 1 , v ] F[i − 1,v] F[i−1,v] ;如果放第 i i i 件物品,那么问题就转化为“前 i − 1 i − 1 i−1 件物品放入剩下的容量为 v − C i v − C_i v−Ci 的背包中”,此时能获得的最大价值就是 F [ i − 1 , v − C i F[i − 1,v − C_i F[i−1,v−Ci ] 再加上通过放入第 i i i 件物品获得的价值 W i W_i Wi 。
代码实现如下:
// N为物品数量,V为最大体积
// c[i]为第i件物品的体积,w[i]为第i件物品的价值
// f[i][j]为前i件物品放入容量为j的背包可以获得的最大价值
for (int i = 1; i <= N; i++) {
for (int j = 0; j <= V; j++) {
f[i][j] = f[i - 1][j];
if (j >= c[i]) // 如果当前背包容量能放下第i件物品
f[i][j] = max(f[i - 1][j], f[i - 1][j - c[i]] + w[i]);
}
}
// 最大价值就是 f[N][V];
优化空间复杂度
以上方法的时间和空间复杂度均为
O
(
V
N
)
O(VN)
O(VN) ,其中时间复杂度应该已经不能再优化
了,但空间复杂度却可以优化到
O
(
V
)
O(V)
O(V) 。
先考虑上面讲的基本思路如何实现,肯定是有一个主循环 i = 1... N i = 1...N i=1...N ,每次算出来二维数组 F [ i , 0... V ] F[i, 0...V] F[i,0...V] 的所有值。那么,如果只用一个数组 F [ 0... V ] F[0...V] F[0...V] ,能不能保证第 i i i 次循环结束后 F [ v ] F[v] F[v] 中表示的就是我们定义的状态 F [ i , v ] F[i, v] F[i,v] 呢? F [ i , v ] F[i, v] F[i,v] 是由 F [ i − 1 , v ] F[i − 1, v] F[i−1,v] 和 F [ i − 1 , v − C i ] F[i − 1, v − C_i ] F[i−1,v−Ci] 两个子问题递推而来,能否保证在推 F [ i , v ] F[i, v] F[i,v] 时(也即在第 i i i 次主循环中推 F [ v ] F[v] F[v] 时)能够取用 F [ i − 1 , v ] F[i − 1, v] F[i−1,v] 和 F [ i − 1 , v − C i ] F[i − 1,v − C_i ] F[i−1,v−Ci] 的值呢?
事实上,这要求在每次主循环中我们以 v = V . . . 0 v = V ...0 v=V...0 的递减顺序计算 F [ v ] F[v] F[v] ,这样才能保证计算 F [ v ] F[v] F[v] 时 F [ v − C i ] F[v − C_i ] F[v−Ci] 保存的是状态 F [ i − 1 , v − C i ] F[i − 1,v − C_i] F[i−1,v−Ci] 的值。
{% note info %}
在二维情况下,状态 f[i][j]
是由上一轮 i - 1
的状态得来的。而优化到一维后,如果我们还是正序,则有可能本应该用第 i - 1
轮的状态却用的是第 i
轮的状态。
举个例子,当我们枚举到第 i = 3
轮,物体的体积为 v[3] = 2
f[8] = max(f[8], f[8 - 2] + w[3])
,这里我们要保证 f[8]
和 f[6]
都是上一轮的状态值。
如果按照正序进行更新 j = 6
时, f[6] = max(f[6], f[6 - 4] + w[3])
,对 f[6]
的状态进行了更新,那么在更新 f[8]
时,用到的 f[6]
就不是上一轮的状态了。
按照逆序的顺序,一维数组的更新顺序为:f[8], f[7], f[6], ...
, 在本轮更新的值,不会影响本轮中其他未更新的值。
{% endnote %}
代码如下:
// N为物品数量,V为最大体积
// c[i]为第i件物品的体积,w[i]为第i件物品的价值
// f[j]为N件物品放入容量为j的背包可以获得的最大价值
for (int i = 1; i <= N; i++) {
for (int j = V; j >= c[i]; j--) {
f[j] = max(f[j], f[j - c[i]] + w[i])
}
}
// 最大价值就是 f[V];
{% note info %}
二维下的状态定义 f[i][j]
是前 i
件物品,背包容量 j
下的最大价值。一维下,少了前 i
件物品这个维度,我们的代码中决策到第 i
件物品(循环到第 i
轮),f[j]
就是前 i
轮已经决策的物品且背包容量 j
下的最大价值。
{% endnote %}
初始化的细节问题
我们看到的求最优解的背包问题题目中,事实上有两种不太相同的问法。有的题目要求 “恰好装满背包” 时的最优解,有的题目则并没有要求必须把背包装满。一种区别这两种问法的实现方法是在初始化的时候有所不同。
如果是第一种问法,要求恰好装满背包,那么在初始化时除了 F [ 0 ] F[0] F[0] 为 0 ,其它 F [ 1.. V ] F[1..V] F[1..V] 均设为 − ∞ -\infty −∞ ,这样就可以保证最终得到的 F [ V ] F[V] F[V] 是一种恰好装满背包的最优解。
如果并没有要求必须把背包装满,而是只希望价格尽量大,初始化时应该将 F [ 0.. V ] F[0..V] F[0..V] 全部设为 0 。
这是为什么呢?可以这样理解:初始化的 F F F 数组事实上就是在没有任何物品可以放入背包时的合法状态。如果要求背包恰好装满,那么此时只有容量为 0 的背包可以在什么也不装且价值为 0 的情况下被“恰好装满”,其它容量的背包均没有合法的解,属于未定义的状态,应该被赋值为 − ∞ -\infty −∞ 了。如果背包并非必须被装满,那么任何容量的背包都有一个合法解 “什么都不装” ,这个解的价值为 0 ,所以初始时状态的值也就全部为 0 了。
这个小技巧完全可以推广到其它类型的背包问题,后面不再对进行状态转移之前的初始化进行讲解。
小结
01 背包问题是最基本的背包问题,它包含了背包问题中设计状态、方程的最基本思想。另外,别的类型的背包问题往往也可以转换成 01 背包问题求解。故一定要仔细体会上面基本思路的得出方法,状态转移方程的意义,以及空间复杂度怎样被优化。
2. 完全背包问题
题目
有 N N N 种物品和一个容量为 V V V 的背包,每种物品都有无限件可用。放入第 i i i 种物品的费用是 C i C_i Ci ,价值是 W i W_i Wi 。求解:将哪些物品装入背包,可使这些物品的耗费的费用总和不超过背包容量,且价值总和最大。
基本思路
这个问题非常类似于 01 背包问题,所不同的是每种物品有无限件。也就是从每种物品的角度考虑,与它相关的策略已并非取或不取两种,而是有取 0 件、取 1 件、取 2 件……直至取 ⌊ V / C i ⌋ ⌊ V / C_i ⌋ ⌊V/Ci⌋ 件等许多种。
如果仍然按照解 01 背包时的思路,令 F [ i , v ] F[i,v] F[i,v] 表示前 i i i 种物品恰放入一个容量为 V V V 的背包的最大权值。仍然可以按照每种物品不同的策略写出状态转移方程,像这样:
F [ i , v ] = m a x { F [ i − 1 , v − k C i ] + k W i ∣ 0 ≤ k C i ≤ V } F[i, v] = max\{F[i-1, v-kC_i] + kW_i|0 \leq kC_i \leq V\} F[i,v]=max{F[i−1,v−kCi]+kWi∣0≤kCi≤V}
这跟 01 背包问题一样有 O ( V N ) O(VN) O(VN) 个状态需要求解,但求解每个状态的时间已经不是常数了,求解状态 F [ i , v ] F[i,v] F[i,v] 的时间是 O ( V / C i ) O(V/C_i) O(V/Ci) ,总的复杂度可以认为是 O ( N V ∑ ( V / C i ) ) O(NV \sum (V/C_i)) O(NV∑(V/Ci)) ,是比较大的。
// N为物品数量,V为最大体积
// c[i]为第i件物品的体积,w[i]为第i件物品的价值
// f[i][j]为N件物品放入容量为j的背包可以获得的最大价值
for (int i = 1; i <= N; i++)
for (int j = 0; j <= V; j++)
for (int k = 0; k * c[i] <= j; k++)
f[i][j] = max(f[i][j], f[i - 1][j - k * c[i]] + k * w[i]);
// 最大价值就是f[N][V];
将 01 背包问题的基本思路加以改进,得到了这样一个清晰的方法。这说明 01 背包问题的方程的确是很重要,可以推及其它类型的背包问题。但我们还是要试图改进这个复杂度。
一个简单有效的优化
完全背包问题有一个很简单有效的优化,是这样的:若两件物品 i i i、 j j j 满足 C i ≤ C j C_i \leq C_j Ci≤Cj 且 W i ≥ W j W_i \geq W_j Wi≥Wj ,则将可以将物品 j j j 直接去掉,不用考虑。
这个优化的正确性是显然的:任何情况下都可将价值小费用高的
j
j
j 换成物美价廉的
i
i
i ,得到的方案至少不会更差。对于随机生成的数据,这个方法往往会大大减少物品的件数,从而加快速度。然而这个并不能改善最坏情况的复杂度,因为有可能特别设计的数据可以一件物品也去不掉。
这个优化可以简单的 O ( N 2 ) O(N^2) O(N2) 地实现,一般都可以承受。另外,针对背包问题而言,比较不错的一种方法是:首先将费用大于 V V V 的物品去掉,然后使用类似计数排序的做法,计算出费用相同的物品中价值最高的是哪个,可以 O ( V + N ) O(V + N) O(V+N) 地完成这个优化。
转化为 01 背包问题求解
01 背包问题是最基本的背包问题,我们可以考虑把完全背包问题转化为 01 背包问题来解。
最简单的想法是,考虑到第 i i i 种物品最多选 ⌊ V / C i ⌋ ⌊V / C_i⌋ ⌊V/Ci⌋ 件,于是可以把第 i i i 种物品转化为 ⌊ V / C i ⌋ ⌊V / C_i⌋ ⌊V/Ci⌋ 件费用及价值均不变的物品,然后求解这个 01 背包问题。这样的做法完全没有改进时间复杂度,但这种方法也指明了将完全背包问题转化为 01 背包问题的思路:将一种物品拆成多件只能选 0 件或 1 件的 01 背包中的物品。
更高效的转化方法是:把第 i i i 种物品拆成费用为 C i ⋅ 2 k C_i \cdot 2^k Ci⋅2k 、价值为 W i ⋅ 2 k W_i \cdot 2^k Wi⋅2k 的若干件物品,其中 k k k 取遍满足 C i ⋅ 2 k ≤ V C_i \cdot 2^k \leq V Ci⋅2k≤V 的非负整数。
这是二进制的思想。因为,不管最优策略选几件第 i i i 种物品,其件数写成二进制后,总可以表示成若干个 2 k 2^k 2k 件物品的和。这样一来就把每种物品拆成 O ( log ⌊ V / C i ⌋ ) O(\log ⌊V / C_i⌋) O(log⌊V/Ci⌋) 件物品,是一个很大的改进。
O(VN) 的算法
这个算法使用一维数组,先看代码:
// N为物品数量,V为最大体积
// c[i]为第i件物品的体积,w[i]为第i件物品的价值
// f[j]为N件物品放入容量为j的背包可以获得的最大价值
for (int i = 1; i <= N; i++)
for (int j = c[i]; j <= V; j++)
f[j] = max(f[j], f[j - c[i]] + w[i]);
// 最大价值就是f[V];
你会发现,这个伪代码与 01 背包问题的伪代码只有 v 的循环次序不同而已。
为什么这个算法就可行呢?首先想想为什么 01 背包中要按照 v v v 递减的次序来循环。让 v v v 递减是为了保证第 i i i 次循环中的状态 F [ i , v ] F[i,v] F[i,v] 是由状态 F [ i − 1 , v − C i ] F[i − 1,v − C_i] F[i−1,v−Ci] 递推而来。换句话说,这正是为了保证每件物品只选一次,保证在考虑 “选入第 i i i 件物品” 这件策略时,依据的是一个绝无已经选入第 i i i 件物品的子结果 F [ i − 1 , v − C i ] F[i − 1,v − C_i] F[i−1,v−Ci] 。而现在完全背包的特点恰是每种物品可选无限件,所以在考虑 “加选一件第 i i i 种物品” 这种策略时,却正需要一个可能已选入第 i i i 种物品的子结果 F [ i , v − C i ] F[i,v − C_i] F[i,v−Ci] ,所以就可以并且必须采用 v v v 递增的顺序循环。这就是这个简单的程序为何成立的道理。
值得一提的是,上面的代码中两层 for
循环的次序可以颠倒。这个结论有可能会带来算法时间常数上的优化。
这个算法也可以由另外的思路得出。例如,将基本思路中求解 F [ i , v − C i ] F[i,v − C_i] F[i,v−Ci] 的状态转移方程显式地写出来,代入原方程中,会发现该方程可以等价地变形成这种形式:
F [ i , v ] = m a x ( F [ i − 1 , v ] , F [ i , v − C i ] + W i ) F[i,v] = max(F[i-1, v], F[i, v-C_i] + W_i) F[i,v]=max(F[i−1,v],F[i,v−Ci]+Wi)
将这个方程用一维数组实现,便得到了上面的代码。
{% note info %}
f [ i , j ] = m a x ( f [ i − 1 , j ] , f [ i − 1 , j − v ] + w , f [ i − 1 , j − 2 ∗ v ] + 2 ∗ w , f [ i − 1 , j − 3 ∗ v ] + 3 ∗ w , . . . . . ) f[i,j] = max( f[i-1,j] , f[i-1,j-v]+w , f[i-1,j-2*v]+2*w , f[i-1,j-3*v]+3*w , .....) f[i,j]=max(f[i−1,j],f[i−1,j−v]+w,f[i−1,j−2∗v]+2∗w,f[i−1,j−3∗v]+3∗w,.....)
f [ i , j − v ] = m a x ( f [ i − 1 , j − v ] , f [ i − 1 , j − 2 ∗ v ] + w , f [ i − 1 , j − 2 ∗ v ] + 2 ∗ w , . . . . . ) f[i , j-v]= max( f[i-1,j-v] , f[i-1,j-2*v] + w , f[i-1,j-2*v]+2*w , .....) f[i,j−v]=max(f[i−1,j−v],f[i−1,j−2∗v]+w,f[i−1,j−2∗v]+2∗w,.....)
由上两式,即可得出上述递推关系:
f [ i , j ] = m a x ( f [ i − 1 , j ] , f [ i , j − v ] + w ) f[i,j] = max(f[i-1, j], f[i, j-v] + w) f[i,j]=max(f[i−1,j],f[i,j−v]+w)
有了上面的关系,那么其实 k
循环可以不要了,之前的代码就可以优化成这样:
for (int i = 1; i <= N; i++)
for (int j = 0; j <= V; j++) {
f[i][j] = f[i - 1][j];
if (j >= c[i])
f[i][j] = max(f[i][j], f[i][j - c[i]] + w[i]);
}
因为和 01 背包代码很相像,我们很容易想到进一步优化成一维数组
for (int i = 1; i <= N; i++)
// 注意,这里的j是从小到大枚举
for (int j = c[i]; j <= V; j++)
f[j] = max(f[j], f[j - c[i]] + w[i]);
{% endnote %}
小结
完全背包问题也是一个相当基础的背包问题,它有两个状态转移方程。希望读者能
够对这两个状态转移方程都仔细地体会,不仅记住,也要弄明白它们是怎么得出来的,最好能够自己想一种得到这些方程的方法。
事实上,对每一道动态规划题目都思考其方程的意义以及如何得来,是加深对动态
规划的理解、提高动态规划功力的好方法。
3. 多重背包问题
题目
有 N N N 种物品和一个容量为 V V V 的背包。第 i i i 种物品最多有 M i M_i Mi 件可用,每件耗费的空间是 C i C_i Ci ,价值是 W i W_i Wi 。求解将哪些物品装入背包可使这些物品的耗费的空间总和不超过背包容量,且价值总和最大。
基本算法
这题目和完全背包问题很类似。基本的方程只需将完全背包问题的方程略微一改
即可。
因为对于第 i i i 种物品有 M i + 1 M_i + 1 Mi+1 种策略:取 0 件,取 1 件……取 M i M_i Mi 件。令 F [ i , v ] F[i,v] F[i,v] 表示前 i i i 种物品恰放入一个容量为 v v v 的背包的最大价值,则有状态转移方程:
F [ i , v ] = m a x { F [ i − 1 , v − k C i ] + k W i ∣ 0 ≤ k ≤ M i } F[i,v] = max\{F[i - 1, v - kC_i] + kW_i|0 \leq k \leq M_i\} F[i,v]=max{F[i−1,v−kCi]+kWi∣0≤k≤Mi}
复杂度是 O ( V ∑ M i ) O(V \sum M_i) O(V∑Mi) 。
// N为物品数量,V为最大体积
// c[i],w[i],m[i]分别为第i件物品的体积,价值,数量
// f[j]为物品放入容量为j的背包可以获得的最大价值
for (int i = 1; i <= N; i++)
for (int j = V; j >= 0; j--)
for (int k = 1; k <= m[i] && k * c[i] <= j; k++)
f[j] = max(f[j], f[j - k * c[i]] + k * w[i]);
// 最大价值就是f[V];
转化为 01 背包问题
另一种好想好写的基本方法是转化为 01 背包求解:把第 i i i 种物品换成 M i M_i Mi 件 01 背包中的物品,则得到了物品数为 ∑ M i \sum M_i ∑Mi 的 01 背包问题。直接求解之,复杂度仍然是 O ( V ∑ M i ) O(V \sum M_i) O(V∑Mi) 。
但是我们期望将它转化为 01 背包问题之后,能够像完全背包一样降低复杂度。仍然考虑二进制的思想,我们考虑把第 i i i 种物品换成若干件物品,使得原问题中第 i i i 种物品可取的每种策略——取 0 ⋯ M i 0 \cdots M_i 0⋯Mi 件——均能等价于取若干件代换以后的物品。另外,取超过 M i M_i Mi 件的策略必不能出现。
方法是:将第
i
i
i 种物品分成若干件 01 背包中的物品,其中每件物品有一个系数。这件物品的费用和价值均是原来的费用和价值乘以这个系数。令这些系数分别为
1
,
2
,
2
2
,
⋯
,
2
k
−
1
,
M
i
−
2
k
+
1
1, 2, 2^2, \cdots, 2^{k−1},M_i − 2^k + 1
1,2,22,⋯,2k−1,Mi−2k+1 ,且
k
k
k 是满足
M
i
−
2
k
+
1
>
0
M_i − 2^k + 1 > 0
Mi−2k+1>0 的最大整数。例如,如果
M
i
M_i
Mi 为 13 ,则相应的
k
=
3
k = 3
k=3 ,这种最多取 13 件的物品应被分成系数分别为 1,2,4,6 的四件物品。
分成的这几件物品的系数和为 M i M_i Mi ,表明不可能取多于 M i M_i Mi 件的第 i i i 种物品。另外这种方法也能保证对于 0 ⋯ M i 0 \cdots M_i 0⋯Mi 间的每一个整数,均可以用若干个系数的和表示。这里算法正确性的证明可以分 0 ⋯ 2 k − 1 0 \cdots 2^{k-1} 0⋯2k−1 和 2 k ⋯ M i 2^k \cdots M_i 2k⋯Mi 两段来分别讨论得出。
这样就将第 i i i 种物品分成了 O ( log M i ) O(\log M_i) O(logMi) 种物品,将原问题转化为了复杂度为 O ( V ∑ log M i ) O(V \sum \log M_i) O(V∑logMi) 的 01 背包问题,是很大的改进。
下面给出 O ( log M ) O(\log M) O(logM) 时间处理一件多重背包中物品的过程:
// N为物品数量,V为最大体积
// c[i],w[i],m[i]分别为第i件物品的体积,价值,数量
// f[j]为物品放入容量为j的背包可以获得的最大价值
struct Good {
int c, w;
};
vector<Good> goods;
// 二进制处理
for (int i = 1; i <= N; i++) {
for (int k = 1; k <= m[i]; k *= 2) {
m[i] -= k;
goods.push_back({k * c[i], k * w[i]})
}
if (m[i] > 0) {
goods.push_back({m[i] * c[i], m[i] * w[i]});
}
}
// 01背包问题
for (auto good : goods)
for (int j = V; j >= good.c; j--)
f[j] = max(f[j], f[j - good.c] + good.w);
// 最大价值就是f[V];
{% note info %}
简单来讲上面的过程就是一个二进制优化的方案
假设有一种物品有 m = 13
个,按照之间的方法我们一共需要枚举 14 次(0,1,2,…,13)
再来看看二进制的方法,我们将 13 分成
1
,
2
,
2
2
,
⋯
,
2
k
−
1
,
M
i
−
2
k
+
1
1, 2, 2^2, \cdots, 2^{k−1},M_i − 2^k + 1
1,2,22,⋯,2k−1,Mi−2k+1,也就是 1,2,4,6
四件物品,这样对于这件物品我们只需要枚举 4 次即可。我们并没有把m
分成 m
份,而是分成了
log
m
\log m
logm 份。
那为什么不是 1,2,4,8
呢?因为如果取了 1,2,4,8
我们实际上不止可以表示出 0-13
,实际上可以表示出 0-15
,但是 14,15
并不是我们需要的。
{% endnote %}
使用单调队列优化
原文中并没有提及使用单调队列优化的算法,在这里给出参考
// q[]为单调队列
// dp_pre表示dp[i-1][]
// dp[]表示dp[i][]
for (int i = 1; i <= N; i++) {
for (int j = 0; j < c[i]; j++) {
// 分别表示队首和队尾
int head = 0, tail = -1;
// k表示当前背包容量
for (int k = j; k <= V; k += c[i]) {
dp[k] = dp_pre[k];
// 维护一个大小为k的区间
// (k - q[head]) / c[i] 表示拿取物品的数量
if (head <= tail && (k - q[head]) / c[i] > m[i])
head++;
// 若队内有值,则该值就是前k个元素的最大值
// dp_pre[q[head]]表示只考虑前i-1个物品时,拿前q[head]个物品的最大价值
if (head <= tail)
dp[k] = max(dp[k], dp_pre[q[head]] + (k - q[head]) / c[i] * w[i]);
// 若队尾元素小于当前元素,则队尾元素出队
// dp_pre[q[tail]] - (q[tail] - j) / c[i] * w[i]表示队尾元素的值
// g[k] - (k - j) / c[i] * w[i]表示当前元素的值
while (head <= tail && dp_pre[q[tail]] - (q[tail] - j) / c[i] * w[i] <= g[k] - (k - j) / c[i] * w[i])
tail--;
// 队列里的元素都比当前元素大,入队
q[++tail] = k;
}
}
}
// 最大价值就是f[V];
小结
在这一讲中,我们看到了将一个算法的复杂度由 O ( V ∑ M i ) O(V \sum M_i) O(V∑Mi) 改进到 O ( V ∑ log M i ) O(V \sum \log M_i) O(V∑logMi) 的过程,还知道了存在复杂度为 O ( V N ) O(VN) O(VN) 的算法。
希望你特别注意 “拆分物品” 的思想和方法,自己证明一下它的正确性,并将完整的程序代码写出来。
4. 混合三种背包问题
问题
如果将前面 1 、 2 、 3 中的三种背包问题混合起来。也就是说,有的物品只可以取一次( 01 背包),有的物品可以取无限次(完全背包),有的物品可以取的次数有一个上限(多重背包)。应该怎么求解呢?
01 背包与完全背包的混合
考虑到 01 背包和完全背包中给出的代码只有一处不同,故如果只有两类物品:一类物品只能取一次,另一类物品可以取无限次,那么只需在对每个物品应用转移方程时,根据物品的类别选用顺序或逆序的循环即可,复杂度是 O ( V N ) O(VN) O(VN) 。
再加上多重背包
如果再加上最多可以取有限次的多重背包式的物品,那么利用单调队列,也可以给
出均摊
O
(
V
N
)
O(VN)
O(VN) 的解法。但如果不考虑单调队列算法的话,用将每个这类物品分成
O
(
log
M
i
)
O(\log M_i)
O(logMi) 个 01 背包的物品的方法也已经很优了。
// N为物品数量,V为最大体积
// c[i],w[i],m[i]分别为第i件物品的体积,价值,数量
// f[j]为物品放入容量为j的背包可以获得的最大价值
// 若 m[i]==-1 则为01背包
// 若 m[i]==0 则为完全背包
// 若 m[i]>0 则为多重背包
struct Thing {
int kind;
int c, w;
};
vector<Thing> things;
for (int i = 1; i <= N; i++) {
if (m[i] < 0) // 01背包
things.push_back({-1, c[i], w[i]});
else if (m[i] == 0) // 完全背包
things.push_back({0, c[i], w[i]});
else { // 多重背包
for (int k = 1; k <= m[i]; k *= 2) { // 二进制优化
m[i] -= k;
things.push_back({-1, k * c[i], k * w[i]});
}
if (m[i] > 0) things.push_back({-1, m[i] * c[i], m[i] * w[i]});
}
}
for (auto thing : things) {
if (thing.king < 0) // 01背包
for (int j = V; j >= thing.c; j--) f[j] = max(f[j], f[j - thing.c] + thing.w);
else // 完全背包
for (int j = thing.c; j <= V; j++) f[j] = max(f[j], f[j - thing.c] + thing.w);
}
// 最大价值就是f[V];
{% note info %}
关键在于将问题分类,将多重背包转化为 01 背包,将三个问题分门别类之后,就可以按照 01 背包,完全背包的思路解决
{% endnote %}
在最初写出这三个过程的时候,可能完全没有想到它们会在这里混合应用。我想这体现了编程中抽象的威力。如果你一直就是以这种“抽象出过程”的方式写每一类背包问题的,也非常清楚它们的实现中细微的不同,那么在遇到混合三种背包问题的题目时,一定能很快想到上面简洁的解法,对吗?
小结
有人说,困难的题目都是由简单的题目叠加而来的。这句话是否公理暂且存之不论,但它在本讲中已经得到了充分的体现。本来 01 背包、完全背包、多重背包都不是什么难题,但将它们简单地组合起来以后就得到了这样一道一定能吓倒不少人的题目。但只要基础扎实,领会三种基本背包问题的思想,就可以做到把困难的题目拆分成简单的题目来解决。
5. 二维费用的背包问题
问题
二维费用的背包问题是指:对于每件物品,具有两种不同的费用,选择这件物品必须同时付出这两种费用。对于每种费用都有一个可付出的最大值(背包容量)。问怎样选择物品可以得到最大的价值。
设第 i i i 件物品所需的两种费用分别为 C i C_i Ci 和 D i D_i Di 。两种费用可付出的最大值(也即两种背包容量)分别为 V V V 和 U U U 。物品的价值为 W i W_i Wi 。
算法
费用加了一维,只需状态也加一维即可。设 F [ i , v , u ] F[i,v,u] F[i,v,u] 表示前 i i i 件物品付出两种费用分别为 v v v 和 u u u 时可获得的最大价值。状态转移方程就是:
F [ i , v , u ] = m a x { F [ i − 1 , v , u ] , F [ i − 1 , v − C i , u − D i ] + W i } F[i,v,u] = max\{F[i-1,v,u], F[i-1,v-C_i,u-D_i]+W_i\} F[i,v,u]=max{F[i−1,v,u],F[i−1,v−Ci,u−Di]+Wi}
如前述优化空间复杂度的方法,可以只使用二维的数组:当每件物品只可以取一次时变量 v v v 和 u u u 采用逆序的循环,当物品有如完全背包问题时采用顺序的循环,当物品有如多重背包问题时拆分物品。
// 以01背包问题为例
// N为物品数量,V为最大体积,U为最大质量
// c[i],d[i],w[i]分别为第i件物品的体积,质量,价值
// f[j][k]为物品放入容量j质量k的背包可以获得的最大价值
for (int i = 1; i <= N; i++)
for (int j = V; j >= c[i]; j--)
for (int k = U; k >= d[i]; k--)
f[j][k] = max(f[j][k], f[j - c[i]][k - d[i]] + w[i]);
// 最大价值就是f[V][U];
{% note info %}
本质上来说和 01 背包问题相同,只不过加了一个维度,这样也就需要我们添加一个新的循环
{% endnote %}
物品总个数的限制
有时,“二维费用” 的条件是以这样一种隐含的方式给出的:最多只能取 U U U 件物品。这事实上相当于每件物品多了一种 “件数” 的费用,每个物品的件数费用均为 1 ,可以付出的最大件数费用为 U U U 。换句话说,设 F [ v , u ] F[v,u] F[v,u] 表示付出费用 v v v 、最多选 u u u 件时可得到的最大价值,则根据物品的类型( 01 、完全、多重)用不同的方法循环更新,最后在 F [ 0... V , 0... U ] F[0...V,0...U] F[0...V,0...U] 范围内寻找答案。
复整数域上的背包问题
另一种看待二维背包问题的思路是:将它看待成复整数域上的背包问题。也就是说,背包的容量以及每件物品的费用都是一个复整数。而常见的一维背包问题则是自然数域上的背包问题。所以说,一维背包的种种思想方法,往往可以应用于二位背包问题的求解中,因为只是数域扩大了而已。
作为这种思想的练习,你可以尝试将后文中提到的“子集和问题”扩展到二维,并试图用同样的复杂度解决。
小结
当发现由熟悉的动态规划题目变形得来的题目时,在原来的状态中加一维以满足新的限制是一种比较通用的方法。希望你能从本讲中初步体会到这种方法。
6. 分组的背包问题
当发现由熟悉的动态规划题目变形得来的题目时,在原来的状态中加一维以满足新的限制是一种比较通用的方法。希望你能从本讲中初步体会到这种方法。
问题
有 N N N 件物品和一个容量为 V V V 的背包。第 i i i 件物品的费用是 C i C_i Ci ,价值是 W i W_i Wi 。这些物品被划分为 K K K 组,每组中的物品互相冲突,最多选一件。求解将哪些物品装入背包可使这些物品的费用总和不超过背包容量,且价值总和最大。
算法
这个问题变成了每组物品有若干种策略:是选择本组的某一件,还是一件都不选。也就是说设 F [ k , v ] F[k,v] F[k,v] 表示前 k k k 组物品花费费用 v v v 能取得的最大权值,则有:
F [ k , v ] = m a x { F [ k − 1 , v ] , F [ k − 1 , v − C i ] + W i } F[k,v] = max\{F[k-1,v], F[k-1,v - C_i] + W_i\} F[k,v]=max{F[k−1,v],F[k−1,v−Ci]+Wi}
// N为物品组数,V为最大体积
// s[i]表示第i组物品的数量
// c[i][j],w[i][j]分别为第i组第j件物品的体积,价值
// f[j]为物品放入容量j的背包可以获得的最大价值
for (int i = 1; i <= N; i++)
for (int j = V; j >= 0; j--)
for (int k = 0; k < s[i]; k++)
if (j >= v[i][k])
f[j] = max(f[j], f[j - c[i][k]] + w[i][k]);
// 最大价值就是f[V];
小结
分组的背包问题将彼此互斥的若干物品称为一个组,这建立了一个很好的模型。不少背包问题的变形都可以转化为分组的背包问题(例如 7 ),由分组的背包问题进一步可定义“泛化物品”的概念,十分有利于解题。
7. 有依赖的背包问题
简化的问题
这种背包问题的物品间存在某种 “依赖” 的关系。也就是说,物品 i i i 依赖于物品 j j j ,表示若选物品 i i i ,则必须选物品 j j j 。为了简化起见,我们先设没有某个物品既依赖于别的物品,又被别的物品所依赖;另外,没有某件物品同时依赖多件物品。
算法
这个问题由 NOIP2006 中 “金明的预算方案” 一题扩展而来。遵从该题的提法,将不依赖于别的物品的物品称为“主件”,依赖于某主件的物品称为“附件”。由这个问题的简化条件可知所有的物品由若干主件和依赖于每个主件的一个附件集合组成。
按照背包问题的一般思路,仅考虑一个主件和它的附件集合。可是,可用的策略非常多,包括:一个也不选,仅选择主件,选择主件后再选择一个附件,选择主件后再选择两个附件……无法用状态转移方程来表示如此多的策略。事实上,设有 n n n 个附件,则策略有 2 n + 1 2^n + 1 2n+1 个,为指数级。
考虑到所有这些策略都是互斥的(也就是说,你只能选择一种策略),所以一个主件和它的附件集合实际上对应于 6 中的一个物品组,每个选择了主件又选择了若干个附件的策略对应于这个物品组中的一个物品,其费用和价值都是这个策略中的物品的值的和。但仅仅是这一步转化并不能给出一个好的算法,因为物品组中的物品还是像原问题的策略一样多。
再考虑对每组内的物品应用 2.3 中的优化。我们可以想到,对于第
k
k
k 个物品组中的物品,所有费用相同的物品只留一个价值最大的,不影响结果。所以,可以对主件
k
k
k 的 “附件集合” 先进行一次 01 背包,得到费用依次为
0
⋯
V
−
C
k
0 \cdots V − C_k
0⋯V−Ck 所有这些值时相应的最
大价值
F
k
[
0
⋯
V
−
C
k
]
F_k [0 \cdots V − C_k]
Fk[0⋯V−Ck] 。那么,这个主件及它的附件集合相当于
V
−
C
k
+
1
V − C_k + 1
V−Ck+1 个物品的物品组,其中费用为
v
v
v 的物品的价值为
F
k
[
v
−
C
k
]
+
W
k
F_k [v − C_k]+W_k
Fk[v−Ck]+Wk ,
v
v
v 的取值范围是
C
k
≤
v
≤
V
C_k \leq v \leq V
Ck≤v≤V 。
也就是说,原来指数级的策略中,有很多策略都是冗余的,通过一次 01 背包后,将主件 k k k 及其附件转化为 V − C k + 1 V − C_k + 1 V−Ck+1 个物品的物品组,就可以直接应用 6 的算法解决问题了。
较一般的问题
更一般的问题是:依赖关系以图论中“森林” 3 的形式给出。也就是说,主件的附件仍然可以具有自己的附件集合。限制只是每个物品最多只依赖于一个物品(只有一个主件)且不出现循环依赖。
解决这个问题仍然可以用将每个主件及其附件集合转化为物品组的方式。唯一不同的是,由于附件可能还有附件,就不能将每个附件都看作一个一般的 01 背包中的物品了。若这个附件也有附件集合,则它必定要被先转化为物品组,然后用分组的背包问题解出主件及其附件集合所对应的附件组中各个费用的附件所对应的价值。
事实上,这是一种树形动态规划,其特点是,在用动态规划求每个父节点的属性之前,需要对它的各个儿子的属性进行一次动态规划式的求值。这已经触及到了“泛化物品”的思想。看完 8 后,你会发现这个“依赖关系树”每一个子树都等价于一件泛化物品,求某节点为根的子树对应的泛化物品相当于求其所有儿子的对应的泛化物品之和。
// 树形dp + 分组背包
// N为物品数量,V为最大体积
// c[i],w[i]分别为第i件物品的体积,价值
// f[i][j]为选择以第i件物品为节点的子树,放入容量j的背包可以获得的最大价值
// vector<int> adj[N]; // 邻接表,每一个节点的儿子
void dfs(int x) {
// 如果要选择以x为根的子树里的节点,则必须选x,设置初始值为w[x]
for (int j = c[x]; j <= V; j++) f[x][j] = w[x];
// 把当前结点x看成是分组背包中的一个组,子节点的每一种选择看作是组内的物品
for (int i = 0; i < adj[x].size(); i++) {
int y = adj[x][i]; // y为x的子节点
dfs(y); // 以y为根再选择结点
for (int j = V - c[x]; j >= 0; j--) {
for (int k = 0; k <= j; k++) { // 遍历子节点的组合
f[x][j] = max(f[x][j], f[x][j - k] + f[y][k]);
}
}
}
}
小结
用物品组的思想考虑那题中极其特殊的依赖关系:物品不能既作主件又作附件,每个主件最多有两个附件,可以发现一个主件和它的两个附件等价于一个由四个物品组成的物品组,这便揭示了问题的某种本质。
8. 泛化物品
定义
考虑这样一种物品,它并没有固定的费用和价值,而是它的价值随着你分配给它的费用而变化。这就是泛化物品的概念。更严格的定义之。在背包容量为 V V V 的背包问题中,泛化物品是一个定义域为 0 ⋯ V 0 \cdots V 0⋯V 中的整数的函数 h h h ,当分配给它的费用为 v v v 时,能得到的价值就是 h ( v ) h(v) h(v) 。
这个定义有一点点抽象,另一种理解是一个泛化物品就是一个数组
h
[
0
⋯
V
]
h[0 \cdots V]
h[0⋯V] ,给它
费用
v
v
v ,可得到价值
h
[
v
]
h[v]
h[v] 。
一个费用为 c c c 价值为 w w w 的物品,如果它是 01 背包中的物品,那么把它看成泛化物品,它就是除了 h ( c ) = w h(c) = w h(c)=w 外,其它函数值都为 0 的一个函数。如果它是完全背包中的物品,那么它可以看成这样一个函数,仅当 v v v 被 c c c 整除时有 h ( v ) = w ⋅ v / c h(v) = w \cdot v/c h(v)=w⋅v/c 其它函数值均为 0 。如果它是多重背包中重复次数最多为 m m m 的物品,那么它对应的泛化物品的函数有 h ( v ) = w ⋅ v / c h(v) = w \cdot v/c h(v)=w⋅v/c 仅当 v v v 被 c c c 整除且 v / c ≤ n v/c \leq n v/c≤n,其它情况函数值均为 0 。
一个物品组可以看作一个泛化物品 h h h 。对于一个 0 ⋯ V 0 \cdots V 0⋯V 中的 v v v ,若物品组中不存在费用为 v v v 的物品,则 h ( v ) = 0 h(v) = 0 h(v)=0 ,否则 h ( v ) h(v) h(v) 取值为所有费用为 v v v 的物品的最大价值。 6 中每个主件及其附件集合等价于一个物品组,自然也可看作一个泛化物品。
泛化物品的和
如果给定了两个泛化物品 h h h 和 l l l ,要用一定的费用从这两个泛化物品中得到最大的价值,这个问题怎么求呢?事实上,对于一个给定的费用 v v v ,只需枚举将这个费用如何分配给两个泛化物品就可以了。同样的,对于 0 ⋯ V 0 \cdots V 0⋯V 中的每一个整数 v v v ,可以求得费用 v v v 分配到 h h h 和 l l l 中的最大价值 f ( v ) f(v) f(v) 。也即
f ( v ) = m a x { h ( k ) + l ( v − k ) ∣ 0 ≤ k ≤ v } f(v)=max\{h(k)+l(v-k)|0 \leq k \leq v\} f(v)=max{h(k)+l(v−k)∣0≤k≤v}
可以看到,这里的 f f f 是一个由泛化物品 h h h 和 l l l 决定的定义域为 0 ⋯ V 0 \cdots V 0⋯V 的函数,也就是说, f f f 是一个由泛化物品 h h h 和 l l l 决定的泛化物品。
我们将 f f f 定义为泛化物品 h h h 和 l l l 的和: h h h 、 l l l 都是泛化物品,若函数 f f f 满足以上关系式,则称 f f f 是 h h h 与 l l l 的和。泛化物品和运算的时间复杂度取决于背包的容量,是 O ( V 2 ) O(V^2) O(V2) 。
由泛化物品的定义可知:在一个背包问题中,若将两个泛化物品代以它们的和,不影响问题的答案。事实上,对于其中的物品都是泛化物品的背包问题,求它的答案的过程也就是求所有这些泛化物品之和的过程。若问题的和为 s s s ,则答案就是 s ( 0 ⋯ V ) s(0 \cdots V) s(0⋯V) 中的最大值。
背包问题的泛化物品
一个背包问题中,可能会给出很多条件,包括每种物品的费用、价值等属性,物品之间的分组、依赖等关系等。但肯定能将问题对应于某个泛化物品。也就是说,给定了所有条件以后,就可以对每个非负整数 v v v 求得:若背包容量为 v v v ,将物品装入背包可得到的最大价值是多少,这可以认为是定义在非负整数集上的一件泛化物品。这个泛化物品——或者说问题所对应的一个定义域为非负整数的函数——包含了关于问题本身的高度浓缩的信息。一般而言,求得这个泛化物品的一个子定义域(例如 0 ⋯ V 0 \cdots V 0⋯V )的值之后,就可以根据这个函数的取值得到背包问题的最终答案。
综上所述,一般而言,求解背包问题,即求解这个问题所对应的一个函数,即该问题的泛化物品。而求解某个泛化物品的一种常用方法就是将它表示为若干泛化物品的和然后求之。
9. 背包问题问法的变换
以上涉及的各种背包问题都是要求在背包容量(费用)的限制下求可以取到的最大价值,但背包问题还有很多种灵活的问法,在这里值得提一下。但是我认为,只要深入理解了求背包问题最大价值的方法,即使问法变化了,也是不难想出算法的。
例如,求解最多可以放多少件物品或者最多可以装满多少背包的空间。这都可以根据具体问题利用前面的方程求出所有状态的值( F F F 数组)之后得到。
还有,如果要求的是“总价值最小”“总件数最小”,只需将状态转移方程中的 max
改成 min
即可。
下面说一些变化更大的问法。
输出方案
一般而言,背包问题是要求一个最优值,如果要求输出这个最优值的方案,可以参照一般动态规划问题输出方案的方法:记录下每个状态的最优值是由状态转移方程的哪一项推出来的,换句话说,记录下它是由哪一个策略推出来的。便可根据这条策略找到上一个状态,从上一个状态接着向前推即可。
还是以 01 背包为例,方程为
F [ i , v ] = m a x { F [ i − 1 , v ] , F [ i − 1 , v − C i ] + W i } F[i, v] = max\{F[i - 1, v], F[i - 1, v - C_i] + W_i\} F[i,v]=max{F[i−1,v],F[i−1,v−Ci]+Wi}
再用一个数组 G [ i , v ] G[i,v] G[i,v] ,设 G [ i , v ] = 0 G[i,v] = 0 G[i,v]=0 表示推出 F [ i , v ] F[i,v] F[i,v] 的值时是采用了方程的前一项(也即 F [ i , v ] = F [ i − 1 , v ] F[i,v] = F[i − 1,v] F[i,v]=F[i−1,v] ), G [ i , v ] = 1 G[i,v] = 1 G[i,v]=1 表示采用了方程的后一项。注意这两项分别表示了两种策略:未选第 i i i 个物品及选了第 i i i 个物品。那么输出方案的代码可以这样写(设最终状态为 F [ N , V ] F[N,V] F[N,V] ):
另外,采用方程的前一项或后一项也可以在输出方案的过程中根据 F [ i , v ] F[i,v] F[i,v] 的值实时地求出来。也即,不须纪录 G G G 数组,将上述代码中的 G [ i , v ] = 0 G[i,v] = 0 G[i,v]=0 改成 F [ i , v ] = F [ i − 1 , v ] F[i,v] = F[i − 1,v] F[i,v]=F[i−1,v] , G [ i , v ] = 1 G[i,v] = 1 G[i,v]=1 改成 F [ i , v ] = F [ i − 1 ] [ v − C i ] + W i F[i,v] = F[i − 1][v − C_i] + W_i F[i,v]=F[i−1][v−Ci]+Wi 也可。
输出字典序最小的最优方案
这里“字典序最小”的意思是 1 ⋯ N 1 \cdots N 1⋯N 号物品的选择方案排列出来以后字典序最小。以输出 01 背包最小字典序的方案为例。
一般而言,求一个字典序最小的最优方案,只需要在转移时注意策略。首先,子问题的定义要略改一些。我们注意到,如果存在一个选了物品 1 的最优方案,那么答案一定包含物品 1 ,原问题转化为一个背包容量为 V − C 1 V − C_1 V−C1 ,物品为 2 ⋯ N 2 \cdots N 2⋯N 的子问题。反之,如果答案不包含物品 1 ,则转化成背包容量仍为 V V V ,物品为 2 ⋯ N 2 \cdots N 2⋯N 的子问题。
不管答案怎样,子问题的物品都是以 i ⋯ N i \cdots N i⋯N 而非前所述的 1 ⋯ i 1 \cdots i 1⋯i 的形式来定义的,所以状态的定义和转移方程都需要改一下。
但也许更简易的方法是,先把物品编号做 x ← N + 1 − x x ← N + 1 − x x←N+1−x 的变换,在输出方案时再变换回来。在做完物品编号的变换后,可以按照前面经典的转移方程来求值。只是在输出方案时要注意,如果 F [ i , v ] = F [ i − 1 , v ] F[i,v] = F[i − 1,v] F[i,v]=F[i−1,v] 和 F [ i , v ] = F [ i − 1 ] [ v − C i ] + W i F[i,v] = F[i − 1][v − C_i] + W_i F[i,v]=F[i−1][v−Ci]+Wi 都成立,应该按照后者来输出方案,即选择了物品 i i i ,输出其原来的编号 N − 1 − i N − 1 − i N−1−i 。
// N为物品数量,V为最大体积
// c[i]为第i件物品的体积,w[i]为第i件物品的价值
// f[i][j]为前i件物品放入容量为j的背包可以获得的最大价值
for (int i = N; i >= 1; i--) {
for (int j = 0; j <= V; j++) {
f[i][j] = f[i + 1][j];
if (j >= c[i])
f[i][j] = max(f[i][j], f[i + 1][j - c[i]] + w[i]);
}
}
int val = V;
for (int i = 1; i <= N; i++) {
if (f[i][val] == f[i + 1][val - c[i]] + w[i]) {
plan.push_back(i);
val -= c[i];
}
}
{% note info %}
由于输出字典序的要求,
需要从第 n
个物品遍历到第 1
个物品,求出当前背包的最大总价值 f[1][V]
再从第 1 个物品遍历到第 n 个物品
若 f[i][j] == f[i + 1][j]
, 则表示 f[i][j]
是从 f[i + 1][j]
状态转移过来的
若 f[i][j] == f[i + 1][j - c[i]] + w[i]
,则表示 f[i][j]
是从 f[i + 1][j - c[i]]
状态转移过来的
{% endnote %}
求方案总数
对于一个给定了背包容量、物品费用、物品间相互关系(分组、依赖等)的背包问题,除了再给定每个物品的价值后求可得到的最大价值外,还可以得到装满背包或将背包装至某一指定容量的方案总数。
对于这类改变问法的问题,一般只需将状态转移方程中的 m a x max max 改成 s u m sum sum 即可。例如若每件物品均是完全背包中的物品,转移方程即为
F [ i , v ] = s u m { F [ i − 1 , v ] , F [ i , v − C i ] } F[i,v] = sum\{F[i-1, v], F[i, v - C_i]\} F[i,v]=sum{F[i−1,v],F[i,v−Ci]}
初始条件是 F [ 0 , 0 ] = 1 F[0,0] = 1 F[0,0]=1 。
事实上,这样做可行的原因在于状态转移方程已经考察了所有可能的背包组成方案。
最优方案的总数
这里的最优方案是指物品总价值最大的方案。以 01 背包为例。
结合求最大总价值和方案总数两个问题的思路,最优方案的总数可以这样求: F [ i , v ] F[i,v] F[i,v] 代表该状态的最大价值, G [ i , v ] G[i,v] G[i,v] 表示这个子问题的最优方案的总数,则在求 F [ i , v ] F[i,v] F[i,v] 的同时求 G [ i , v ] G[i,v] G[i,v] 的代码如下:
// N为物品数量,V为最大体积
// c[i]为第i件物品的体积,w[i]为第i件物品的价值
// f[j]为物品放入容量为j的背包可以获得的最大价值
// g[j]为背包容积为j时总价值为最佳的方案数(初始化为1)
for (int i = 1; i <= N; i++) {
for (int j = V; j >= c[i]; j--) {
int val = f[j - c[i]] + w[i]; // 选择当前物品后的最大价值
if (f[j] == val)
g[j] = g[j] + g[j - c[i]]; // 若相等,方案数为两者之和
else if (val > f[j])
g[j] = g[j - c[i]]; // 若最大价值增加,则更新为选择第i个物品的方案
f[j] = max(f[j], f[j - c[i]] + w[i]);
}
}
// 最大方案数就是 g[V];
{% note info %}
f[j]
为物品放入容量为 j
的背包可以获得的最大价值
g[j]
为背包容积为 j
时总价值为最佳的方案数
先初始化所有的 g[i]
为 1
,因为背包里什么也不装也是一种方案
如果装新物品的方案总价值更大,那么 g[j] = g[j - c[i]]
如果总价值相等,方案数为两者之和
{% endnote %}
求次优解、第 K 优解
对于求次优解、第 K K K 优解类的问题,如果相应的最优解问题能写出状态转移方程、用动态规划解决,那么求次优解往往可以相同的复杂度解决,第 K K K 优解则比求最优解的复杂度上多一个系数 K K K 。
其基本思想是,将每个状态都表示成有序队列,将状态转移方程中的 max/min
转化成有序队列的合并。
这里仍然以 01 背包为例讲解一下。
首先看 01 背包求最优解的状态转移方程:
F [ i , v ] = m a x { F [ i − 1 , v ] , F [ i − 1 , v − C i ] + W i } F[i, v] = max\{F[i - 1, v], F[i - 1, v - C_i] + W_i\} F[i,v]=max{F[i−1,v],F[i−1,v−Ci]+Wi}
如果要求第 K K K 优解,那么状态 F [ i , v ] F[i,v] F[i,v] 就应该是一个大小为 K K K 的队列 F [ i , v , 1 ⋯ K ] F[i,v,1 \cdots K] F[i,v,1⋯K] 。其中 F [ i , v , k ] F[i,v,k] F[i,v,k] 表示前 i i i 个物品中,背包大小为 v v v 时,第 k k k 优解的值。这里也可以简单地理解为在原来的方程中加了一维来表示结果的优先次序。显然 F [ i , v , 1 ⋯ K ] F[i,v,1 \cdots K] F[i,v,1⋯K] 这 K K K 个数是由大到小排列的,所以它可看作是一个有序队列。
然后原方程就可以解释为:
F
[
i
,
v
]
F[i,v]
F[i,v] 这个有序队列是由
F
[
i
−
1
,
v
]
F[i − 1,v]
F[i−1,v] 和
F
[
i
−
1
,
v
−
C
i
]
+
W
i
F[i − 1,v − C_i] + W_i
F[i−1,v−Ci]+Wi 这两个有序队列合并得到的。前者
F
[
i
−
1
]
[
v
]
F[i − 1][v]
F[i−1][v] 即
F
[
i
−
1
,
v
,
1
⋯
K
]
F[i − 1,v,1 \cdots K]
F[i−1,v,1⋯K] ,后者
F
[
i
−
1
,
v
−
C
i
]
+
W
i
F[i − 1,v − C_i] + W_i
F[i−1,v−Ci]+Wi 则理解为在
F
[
i
−
1
,
v
−
C
i
,
1
⋯
K
]
F[i − 1,v − C_i ,1 \cdots K]
F[i−1,v−Ci,1⋯K] 的每个数上加上
W
i
W_i
Wi 后得到的有序队列。合并这两个有序队列并将结果的前
K
K
K 项储存到
F
[
i
,
v
,
1
⋯
K
]
F[i,v,1 \cdots K]
F[i,v,1⋯K] 中的复杂度是
O
(
K
)
O(K)
O(K) 。最后的第
K
K
K 优解的答案是
F
[
N
,
V
,
K
]
F[N,V,K]
F[N,V,K] 。总的时间复杂度是
O
(
V
N
K
)
O(VNK)
O(VNK) 。
为什么这个方法正确呢?实际上,一个正确的状态转移方程的求解过程遍历了所有可用的策略,也就覆盖了问题的所有方案。只不过由于是求最优解,所以其它在任何一个策略上达不到最优的方案都被忽略了。如果把每个状态表示成一个大小为
K
K
K 的数组,并在这个数组中有序地保存该状态可取到的前
K
K
K 个最优值。那么,对于任两个状态的 max
运算等价于两个由大到小的有序队列的合并。
另外还要注意题目对于“第 K K K 优解”的定义,是要求将策略不同但权值相同的两个方案是看作同一个解还是不同的解。如果是前者,则维护有序队列时要保证队列里的数没有重复的。
小结
显然,这里不可能穷尽背包类动态规划问题所有的问法。甚至还存在一类将背包类动态规划问题与其它领域(例如数论、图论)结合起来的问题,在这篇论背包问题的专文中也不会论及。但只要深刻领会前述所有类别的背包问题的思路和状态转移方程,遇到其它的变形问题,应该也不难想出算法。
触类旁通、举一反三,应该也是一个程序员应有的品质吧。