Tối ưu hóa bất đẳng thức tứ giác
四边形不等式优化利用的是状态转移方程中的决策单调性,也常称为 决策单调性优化 DP.
基础知识
考虑最简单的情形,我们要解决如下一系列最优化问题:
这里假定成本函数 \(w(j,i)\) 可以在 \(O(1)\) 时间内计算.
约定
动态规划的状态转移方程经常可以写作一系列最优化问题的形式.以(1)式为例,这些问题含有参数 \(i\),问题的目标函数和可行域都可以依赖于 \(i\).每一个问题都是在给定参数 \(i\) 时,选取某个可行解 \(j\) 来最小化目标函数的取值.为表述方便,下文将参数为 \(i\) 的最优化问题简称为「问题 \(i\)」,该最优化问题的可行解 \(j\) 称为「决策 \(j\)」,目标函数在最优解处取得的值则称为「状态 \(f(i)\)」.同时,记问题 \(i\) 对应的最小最优决策点为 \(\operatorname{opt}(i)\).
在一般的情形下,这些问题总时间复杂度为 \(O(n^2)\).这是由于对于问题 \(i\),我们需要考虑所有可能的决策 \(j\).而在满足决策单调性时,可以有效缩小决策空间,优化总复杂度.
- 决策单调性:对于任意 \(i_1 < i_2\),必然成立 \(\operatorname{opt}(i_1) \leq \operatorname{opt}(i_2)\).
附注
对于问题 \(i\),最优决策集合未必是一个区间.决策单调性实际可以定义在最优决策集合上.对于集合 \(A\) 和 \(B\),可以定义 \(A \leq B\) 当且仅当对于任意 \(a\in A\) 和 \(b\in B\),成立 \(\min\{a,b\}\in A\) 和 \(\max\{a,b\}\in B\).这蕴含最小(最大)最优决策点的单调性,即此处采取的定义.本文关于最小最优决策点叙述的结论,同样适用于最大最优决策点.但是,存在情形,某更大问题的最小最优决策严格小于另一更小问题的最大最优决策,亦即可能对某些 \(i_1 < i_2\) 成立 \(\mathop{\mathrm{optmax}}(i_1) > \mathop{\mathrm{optmin}}(i_2)\),所以在书写代码时,应保证总是求得最小或最大的最优决策点.
另一方面,拥有相同最小最优决策的问题构成一个区间.这一区间,作为最小最优决策的函数,应严格递增.亦即,给定 \(j_1 = \operatorname{opt}(i_1)\),\(j_2 = \operatorname{opt}(i_2)\),如果 \(j_1 < j_2\),那么必然有 \(i_1 < i_2\).换言之,如果决策 \(j_1 < j_2\) 能够成为最小最优决策的问题区间分别是 \([l_{j_1},r_{j_1}]\) 和 \([l_{j_2},r_{j_2}]\),那么必然有 \(r_{j_1} < l_{j_2}\).
最常见的判断决策单调性的方法是通过四边形不等式(quadrangle inequality).在不同的语境下,这一性质也常称为 Monge 性质(用于描述矩阵 \(A_{j,i}\))或次模性(submodularity,用于描述以区间为自变量的函数 \(f([j,i])\)).
-
四边形不等式:如果对于任意 \(a\leq b\leq c\leq d\) 均成立
\[ w(a,c)+w(b,d) \leq w(a,d)+w(b,c), \]则称函数 \(w\) 满足四边形不等式(简记为「交叉小于包含」).若等号永远成立,则称函数 \(w\) 满足 四边形恒等式.
如果没有特别说明,以下都会保证 \(a\leq b\leq c\leq d\).四边形不等式给出了一个决策单调性的充分不必要条件.
定理 1
若 \(w\) 满足四边形不等式,则问题 (1) 满足决策单调性.
证明
要证明这一点,可采用反证法.假设对某些 \(c < d\),成立 \(a = \operatorname{opt}(d) < \operatorname{opt}(c) = b\).此时有 \(a < b \leq c < d\).根据最优化条件,\(w(a,d) \leq w(b,d)\) 且 \(w(b,c) < w(a,c)\),于是,\(w(a,d) - w(b,d) \leq 0 < w(a,c) - w(b,c)\),这与四边形不等式矛盾.
四边形不等式可以理解在合理的定义域内,\(w\) 的二阶混合差分 \(\Delta_i\Delta_jw(j,i)\) 非正.
利用决策单调性,有很多常见算法都可以将算法复杂度优化到 \(O(n\log n)\).这些算法的适用范围、实现难度、运行效率各不相同,需要根据实际场景选择合适的算法.这主要取决于 \(w(j,i)\) 的性质.不加说明时,本文默认 \(w(i,j)\) 可以 随机访问,即 \(w(j,i)\) 可以在 \(O(1)\) 时间内查询或计算.但是,并非所有问题中,\(w(j,i)\) 都这样容易计算.因此,除了基本情形外,本文还讨论了 \(w(j,i)\) 只具有如下性质时,利用决策单调性优化 DP 的方法:
- 移动访问:\(w(j,i)\) 可以从 \(w(j\pm 1,i)\) 或 \(w(j,i\pm 1)\) 以 \(O(1)\) 时间转移得到.(类似 莫队算法 中的情形)
- 动态计算:\(w(j,i)\) 的计算依赖于 \(\{f(j'):j' < j\}\).这意味着 \(f\) 和 \(w\) 只能顺次计算.下文介绍了不限制区间个数的区间分拆问题,它就属于这一情形.
这两条性质并不互斥,可能存在 \(w(j,i)\) 既需要动态计算,又只支持移动访问的情形.
分治
要求解所有状态,只需要求解所有最优决策点.为了对所有 \(1 \leq i \leq n\) 求解 \(\operatorname{opt}(i)\),首先计算 \(\operatorname{opt}(n/2)\),而后分别计算 \(1 \leq i < n/2\) 和 \(n/2 < i \leq n\) 上的 \(\operatorname{opt}(i)\),注意此时已知前半段的 \(\operatorname{opt}(i)\) 必然位于 \(1\) 和 \(\operatorname{opt}(n/2)\) 之间(含端点),而后半段的 \(\operatorname{opt}(i)\) 必然位于 \(\operatorname{opt}(n/2)\) 和 \(n\) 之间(含端点).对于两个子区间,也类似处理,直至计算出每个问题的最优决策.在分治的过程中记录搜索的上下边界,就可以保证算法复杂度控制在 \(O(n\log n)\).递归树层数为 \(O(\log n)\),而每层中,单个决策点至多计算两次,所以总的计算次数是 \(O(n\log n)\).
参考实现
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | |
除了随机访问的基本情形外,分治算法还可以应用于 \(w(j,i)\) 只支持移动访问的情形.只需要在计算过程中维护游标 \((j,i)\) 和相应函数值 \(w(j,i)\),在需要查询新的值时,将游标暴力移动到当前位置并更新函数值即可.这样做的时间复杂度仍然为 \(O(n\log n)\).对此更为详细的讨论,可以参考下文 简化 LARSCH 算法 一节.但是,分治算法无法解决 \(w(j,i)\) 需要动态计算的情形,因为分治算法没有办法在左半区间问题尚未解决时,就计算出区间中点处的最小最优决策 \(\operatorname{opt}(n/2)\).
二分队列
注意到对于每个决策点 \(j\),能使其成为最小最优决策点的问题 \(i\) 必然构成一个区间.可以通过单调队列记录到目前为止每个决策点可以解决的问题的区间,这样,问题的最优解自然可以通过队列中记录的决策点计算得到.
具体地,算法需要顺次遍历决策点.当遍历到决策点 \(k\) 时,队列中需要记录到目前为止每个可行的决策点 \(j\) 和能够解决的问题区间左右端点 \(l_j\) 和 \(r_j\) 构成的 三元组.对于给定区间 \([l_j,r_j]\) 内的问题,\(j\) 应该是到目前为止考虑过的决策点(即区间 \([1,k]\) 中的决策点)中最小最优的.每时每刻,队列中存储的决策未必是连续的,但是尚未解决的问题 \([j,n]\) 应该是队列中存储的问题区间的不交并.
为了说明队列更新过程中,决策点 \(j\) 是最小最优决策的问题 \(i\) 总构成一段连续的区间,需要适当加强前文的结论:
推论 1
设 \(\operatorname{opt}_k(i)\) 是仅考虑 \([1,k]\) 中的决策时,问题 \(i\) 的最小最优决策.如果 \(w\) 满足四边形不等式,那么对于任意 \(i_1 < i_2\),必然成立 \(\operatorname{opt}_k(i_1) \leq \operatorname{opt}_k(i_2)\).
证明
设 \(M\) 是充分大的正实数.函数 \(w'(j,i) = w(j,i) + M[j > k]\) 仍然满足四边形不等式,其中,\([\cdot]\) 是 Iverson 括号.考虑以 \(w'\) 为成本函数的辅助 DP.在辅助 DP 中,对于任何问题 \(i\),决策 \(j > k\) 都不可能是最小最优的,即 \(\operatorname{opt}'(i) = \operatorname{opt}'_k(i) = \operatorname{opt}_k(i)\).对辅助 DP 应用定理 1 就得到本推论.
该算法过程如下:1
- 初始时,队列是空的.类似于单调队列,每次考虑下一个决策 \(j\) 的时候,都需要进行出队和入队操作.
- 出队:首先将上一个问题 \(j-1\) 从队列中移除.如果队首的决策能够解决的问题的右端点恰为 \(j-1\),直接弹出队首;否则,将队首决策能够解决问题的左端点更新为 \(j\).
- 入队:要对决策 \(j\) 进行入队时,首先比较它和队尾的决策 \(j'\).
- 如果对于问题 \(l_{j'}\),将要入队的决策 \(j\) 比已有的决策 \(j'\) 严格更优,即 \(w(j,l_{j'}) < w(j',l_{j'})\) 时,则弹出队尾的决策 \(j'\).此操作持续到队列为空或队尾的决策 \(j'\) 比起 \(j\) 对于问题 \(l_{j'}\) 更优时为止.
- 如果队列已空,入队 \((j,j,n)\),即认为决策 \(j\) 是尚未解决的所有问题的最优解.
- 如果队尾决策 \(j'\) 对于问题 \(r_{j'}\) 同样不劣于将入队的决策 \(j\),那么当 \(r_{j'} < n\) 时,入队 \((j,r_{j'}+1,n)\),表示 \(j\) 是问题 \([r_{j'}+1,n]\) 的最小最优决策;否则,不需要入队 \(j\),因为它并不比已有的决策更优.
- 最后的情形是,队尾决策 \(j'\) 比起要入队的决策 \(j\) 对于问题 \(l_{j'}\) 严格更优,而对于问题 \(r_{j'}\) 严格更劣.这说明,存在问题 \(i\in(l_{j'},r_{j'}]\) 使得问题 \([l_{j'},i-1]\) 的最小最优决策为 \(j'\) 且问题 \([i,r_{j'}]\) 的最小最优决策为 \(j\).因而,需要通过 二分 找到最小的 \(i\in[l_{j'},r_{j'}]\) 使得 \(w(j,i) < w(j',i)\),再将队尾的区间右端点 \(r_{j'}\) 修改为 \(i-1\),并入队 \((j,i,n)\).
- 处理完决策 \(j\) 后,就已经处理了所有到 \(j\) 为止的决策.此时,队首决策就是问题 \(j\) 的最小最优决策,可以记录相应的最优解.
参考实现
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | |
类似于单调队列,每个决策点至多入队一次,出队一次.其中,出队是 \(O(1)\) 的,而入队是 \(O(\log n)\) 的(可能需要二分),所以总的时间复杂度是 \(O(n\log n)\).
由于二分队列算法顺次考虑所有问题和决策点,它可以应用于 \(w(j,i)\) 需要动态计算的情形.这是它相较于分治算法的优势.但是,因为算法中的二分步骤依赖于对 \(w(j,i)\) 的随机访问,它无法应用于 \(w(j,i)\) 只支持移动访问的情形.
例题 1:「POI2011」Lightning Conductor
给定一个长度为 \(n\) 的序列 \(a_1,a_2,\cdots,a_n\),要求对于每一个 \(1 \leq i \leq n\),找到最小的非负整数 \(f_i\) 满足
思路
显然,经过不等式变形,我们可以得到待求整数 \(f_i = \max_{j}\{a_j+\sqrt{|i-j|}-a_i\}\).不妨先考虑 \(j \leq i\) 的情况(另外一种情况类似),此时我们可以得到状态转移方程:
根据 \(-\sqrt{x}\) 的凸性,我们很容易得出(后文将详细描述)函数 \(w(l, r) = -a_l - \sqrt{r-l} + a_r\) 满足四边形不等式,因此套用上述的算法便可在 \(O(n\log n)\) 的时间内解决此题了.
实现
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | |
简化 LARSCH 算法
前两种算法都无法处理 \(w(j,i)\) 既需要动态计算,也只支持移动访问的情形.本节介绍一种能够同时克服这两种困难的算法.它是 Larmore 和 Schieber 在 1991 年提出的 LARSCH 算法2的简化版本,故称为 简化 LARSCH 算法.该算法的原始版本可以在 \(O(n)\) 时间内解决决策单调性 DP 问题,但实现较复杂,本文不做介绍.
仍然考虑分治求解.求解区间 \((l,r]\) 内的问题时,假设如下信息已知:
- 区间 \([1,l]\) 中问题 \(i\) 的最小最优决策 \(\operatorname{opt}(i)\) 和最优值,以及
- 仅考虑区间 \([1,l]\) 中的决策,问题 \(r\) 的最小最优决策 \(\operatorname{opt}_l(r)\) 和最优值.
区间 \((l,r]\) 内的问题求解结束时,需要得到区间 \((l,r]\) 内问题的最小最优决策和最优值.
设 \(\textit{mid}\) 为区间 \((l,r]\) 的中点.求解过程如下:
- 遍历决策 \(i\in[\operatorname{opt}(l),\operatorname{opt}_l(r)]\),更新问题 \(\textit{mid}\) 的最小最优决策和最优值.
- 递归求解区间 \((l,\textit{mid}]\) 中的问题.
- 遍历决策 \(i\in(l,\textit{mid}]\),更新问题 \(r\) 的最小最优决策和最优值.
- 递归求解区间 \((\textit{mid},r]\) 中的问题.
对整个区间 \([1,n]\) 执行递归前,首先需要用决策 \(j=1\) 更新问题 \(i\in\{1,n\}\).该算法在递归到 \(l=r\) 时就终止.
参考实现
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | |
首先,可以说明这一算法的正确性.为此,只要检查每一递归求解的步骤(即步骤 2 和 4)前都满足上文给出的前提条件.由前一节得到的推论 1,有 \(\operatorname{opt}(l)=\operatorname{opt}_l(l)\le\operatorname{opt}_l(\textit{mid})\le\operatorname{opt}_l(r)\),所以步骤 1 后,\(\operatorname{opt}_l(\textit{mid})\) 是已知的,进而执行步骤 2 前递归求解区间 \((l,\textit{mid}]\) 中问题的前提条件成立.由于之前 \(\{\operatorname{opt}(i):i\in[1,l]\}\) 是已知的,步骤 2 又得到了 \(\{\operatorname{opt}(i):i\in(l,\textit{mid}]\}\) 的取值,所以这之后 \(\{\operatorname{opt}(i):i\in[1,\textit{mid}]\}\) 都是已知的;同时,由于之前 \(\operatorname{opt}_l(r)\) 是已知的,步骤 3 后,\(\operatorname{opt}_\textit{mid}(r)\) 就也是已知的.因此,执行步骤 4 前递归求解区间 \((\textit{mid},r]\) 中问题的前提条件也成立.
然后,需要说明该算法的复杂度仍然是 \(O(n\log n)\) 的.递归树的层数是 \(O(\log n)\) 的.对于递归树中的同一层的每个结点,分别遍历区间 \([\operatorname{opt}(l),\operatorname{opt}_l(r)]\) 和 \((l,\textit{mid}]\) 中的决策.因为 \(\operatorname{opt}(l)\le\operatorname{opt}_l(r)\le\operatorname{opt}(r)\),所以在同一层中,每个决策点只遍历了 \(O(1)\) 次.故而,递归树的每一层中总的遍历次数是 \(O(n)\) 的.假设单次访问或计算 \(w(j,i)\) 的复杂度是 \(O(1)\) 的,算法的时间复杂度就是 \(O(n\log n)\) 的.
对于一些 \(w(j,i)\) 只支持移动访问的情形,这一算法的复杂度仍然是 \(O(n\log n)\) 的.此时,需要为算法流程中的步骤 1 和步骤 3 分别维护游标 \((j,i)\) 和 \(w(j,i)\) 的当前取值.每次需要访问新的值时,需要将游标 \((j,i)\) 从上一次访问时的位置暴力更新到当前位置,并对函数值 \(w(j,i)\) 进行转移.容易验证,当对递归树进行遍历时,这些暴力更新的总次数是 \(O(n\log n)\) 的.所以,算法的时间复杂度仍然是 \(O(n\log n)\) 的.
非随机访问的复杂度证明
只需要证明游标移动的总次数是 \(O(n\log n)\) 的.设 \(A\) 和 \(B\) 分别为步骤 1 和 3 对应的游标.实际上可以保证:在求解区间 \((l,r]\) 的问题之前,游标 \(A\) 处于位置 \((\operatorname{opt}(l),l)\),游标 \(B\) 处于位置 \((l,l)\);而在这之后,游标 \(A\) 处于位置 \((\operatorname{opt}(r),r)\),游标 \(B\) 处于位置 \((r,r)\).
考虑构造如下游标移动规则.步骤 1 中,可以令游标 \(A\) 沿着路径
移动.此时,游标 \(A\) 和 \(B\) 均处于求解区间 \((l,\textit{mid}]\) 的问题之前的规定位置上.步骤 2 中,按规定,游标 \(A\) 将移动到 \((\operatorname{opt}(\textit{mid}),\textit{mid})\),游标 \(B\) 将移动到 \((\textit{mid},\textit{mid})\).步骤 3 中,可以令游标 \(B\) 沿着路径
移动.此时,游标 \(A\) 和 \(B\) 均处于求解区间 \((\textit{mid},r]\) 的问题之前的规定位置上.步骤 4 中,按规定,游标 \(A\) 将移动到 \((\operatorname{opt}(r),r)\),游标 \(B\) 将移动到 \((r,r)\).两个游标均在结束求解区间 \((l,r]\) 的问题的规定位置上.因此,这一移动规则符合上述规定.而且,该移动规则足以完成步骤 1 和 3 中的所有计算.直接计算该规则中游标的移动次数可知,步骤 1 需要
次移动,步骤 3 需要
次移动.将这些移动次数对递归树中的所有结点求和,利用同一层中的所有 \([l,r]\) 和 \([\operatorname{opt}(l),\operatorname{opt}_l(r)]\) 至多只在端点处重合这一性质,就可以说明总的移动次数是 \(O(n\log n)\) 的.
由于上述移动规则比起实际计算时游标的移动设置了更多的途径点,所以游标的实际移动次数不会超过该规则下移动次数的估计.因此,游标的实际移动次数也是 \(O(n\log n)\) 的.
由于该算法在求解区间 \((l,r]\) 的问题前,已经计算出了区间 \([1,l]\) 中的最优解 \(f(i)\),所以该算法也可以应用于 \(w(j,i)\) 需要动态计算的情形.
区间分拆问题
考虑将某个区间拆分成若干个子区间的问题.形式化地说,将给定区间 \([1,n]\) 拆分成 \([a_1,b_1],\cdots,[a_k,b_k]\),其中,\(a_1=1\),\(b_k=n\),以及 \(b_{i}+1=a_{i+1}\) 对任意 \(i < k\) 都成立.对于给定拆分,成本为 \(\sum_{i=1}^kw(a_i,b_i)\).问题要求最小化这一成本.可以列出如下的 1D1D 状态转移方程.
这里,\(f(0)=0\).注意到,只要 \(w(j,i)\) 满足四边形不等式,\(f(j-1)+w(j,i)\) 必然满足四边形不等式,因为第一项并不包括 \(j\) 和 \(i\) 的交叉项,在混合差分时会消去.但是由于成本函数依赖于前面的子问题,这一转移只能够顺序计算,所以无法应用前文描述的第一种分治算法,通常只适合应用二分队列算法或简化 LARSCH 算法.算法复杂度为 \(O(n\log n)\).
限制区间个数的情形
上述问题可以加强为限制区间个数的情形,即问题指定将区间拆分成 \(m\) 个子区间.此时需要将拆分后的区间个数作为转移状态的一维.相应地,有 2D1D 状态转移方程如下.
这里,\(f(0,0)=0\),\(f(0,i)=f(k,0)=\infty\) 对任意 \(1\leq k\leq m\) 和 \(1\leq i\leq n\) 都成立.和上文同样的道理,这里的 \(f(k-1,j-1)+w(j,i)\) 必然满足四边形不等式.此时对于第 \(i\) 层的计算,并不再依赖于该层的结果,所以对于每一层,都可以通过上一节描述的任何算法进行计算,此时算法复杂度为 \(O(mn\log n)\).
对于这一问题,利用决策单调性,实际上还存在其他的优化算法.第二种优化思路依赖于如下结果.这种优化算法和下文详细描述的 Knuth 优化算法十分相似.
定理 2
若 \(w\) 满足四边形不等式,则对于问题 (2) 成立 \(\operatorname{opt}(k-1,i) \leq \operatorname{opt}(k,i) \leq \operatorname{opt}(k,i+1)\).
证明
第二个不等式只是第 \(k\) 层的决策单调性.关键在于第一个不等式.
下证 \(\operatorname{opt}(k,i) \leq \operatorname{opt}(k+1,i)\).假设有如下两个区间 \([1,i]\) 的分划(逆序标号):\([a_{k},d_{k}],\cdots,[a_1,d_1]\) 和 \([b_{k+1},c_{k+1}],\cdots,[b_1,c_1]\).其中,每个区间的左端点都是其右端点处对应问题的最小最优决策;同样地,从右向左考虑所有可能的分划,右端点也是左端点对应问题的最小最优决策.例如,\(d_j\) 和 \(c_j\) 分别是将 \([a_j,i]\) 和 \([b_j,i]\) 分成 \(j\) 段左起第一个区间右端点的最小最优决策.根据决策单调性,如果 \(a_{j-1} > b_{j-1}\),亦即 \(d_j > c_j\),那么必然有 \(a_j > b_j\).由此,如果所证不成立,则有 \(a_1 > b_1\).进而可以归纳地证明 \(a_{k} > b_{k}\).这显然与所设矛盾.由此得证.
第一个不等式可以另证如下.同样考虑上面证明中的两个分划.如果所证命题不成立,则有 \(a_1 > b_1\),但是由于有 \(a_{k} < b_{k}\),我们可以找到最小的 \(j>1\) 使得 \(a_j \leq b_j\).进而,此时有 \(a_{j-1} > b_{j-1}\),故 \(d_j>c_j\).我们找到了一组区间满足 \(a_j \leq b_j \leq c_j < d_j\).考虑将这两个分拆重新组合的结果.考虑分拆 \([b_{k+1},c_{k+1}],\cdots,[b_{j+1},c_{j+1}],[b_j,d_j],[a_{j-1},d_{j-1}],\cdots,[a_1,d_1]\),共 \((k+1)\) 段,于是由前设的最优性可推知,
同样地,考虑分拆 \([a_{k},d_{k}],\cdots,[a_{j+1},d_{j+1}],[a_j,c_j],[b_{j-1},c_{j-1}],\cdots,[b_1,c_1]\),共 \(k\) 段,则有
此时,不等号是严格的,因为 \(a_1 > b_1\),但是按假设,\(a_1\) 是所有 \(k\) 段分拆最末一段的左端点中最小最优的.两个不等式条件相加,得到 \(w(b_j,c_j) + w(a_j,d_j) < w(b_j,d_j) + w(a_j,c_j)\),这有悖于四边形不等式.故而原结论得证.
利用这一结果,我们可以限制决策 \(j\) 的搜索范围.算法实现时,对 \(k\) 正向遍历,对 \(i\) 逆向遍历,在之前已确定的上下界范围内暴力搜索 \(j\) 就可以保证 \(O(n(n+m))\) 的算法复杂度.
注意
这里算法复杂度不是 \(O(nm)\) 的.正确的复杂度计算需要考虑 \(n\times m\) 维状态矩阵.因为对于问题 \((i,k)\) 只需要考虑 \(\operatorname{opt}(k-1,i) \leq j \leq \operatorname{opt}(k,i+1)\) 中的决策,所以每条次对角线上(即 \(i-k\) 为一定值)的问题所需遍历的决策总数为 \(O(n)\) 的.这样的对角线共计 \((n+m)\) 条,故而总的时间复杂度为 \(O(n(n+m))\).
最后一种优化方法来源于如下的观察.
定理 3
若 \(w\) 满足四边形不等式,则问题 (2) 的最优解 \(g(k):=f(n,k)\) 是关于 \(k\) 的凸函数.
证明
下证 \(g(k-1) + g(k+1) \ge 2g(k)\).为此,考虑长度为 \((k-1)\) 段和 \((k+1)\) 段的最优分划,分别是 \([a_1,d_1],\cdots,[a_{k-1},d_{k-1}]\) 和 \([b_1,c_1],\cdots,[b_{k+1},c_{k+1}]\).取最小的 \(1 \leq j \leq k-1\) 使得 \(c_{j+1} \leq d_j\),其存在性可由 \(c_{k} < n = d_{k-1}\) 推知.根据其最小性得知,\(b_{j+1} > a_j\).所以,\(a_j < b_{j+1} \leq c_{j+1} \leq d_j\).与上文类似,交换两个现有分拆的后半段,可以得到如下两个区间分拆:
两个所得区间都是 \(k\) 段的,所以由最优性条件可知
这里第二个不等式正是四边形不等式.所求凸性由此得证.
这一结论保证了可以通过 WQS 二分(国外称 Aliens Trick)的方法解决此问题.具体来说,考虑带参的成本函数 \(w_c(j,i):=w(j,i)+c\),解决不限制区间个数的问题,求得其最优解为 \(f_c(n)\).随着实数 \(c\) 递增,相应的最优区间的数目单调递减,故而可以通过二分的方法找到恰使得最优区间个数等于 \(m\) 的参数 \(c\),则原题最优解为 \(f(n,m) = f_c(n)-cm\).这里的实数 \(c\) 可以看作区间个数限制的 Lagrange 乘子.该算法的实现有很多细节,可以参考 WQS 二分 页面.这一算法的时间复杂度为 \(O(n\log n\log C)\),这里 \(C\) 为某一常数.
对于限制区间个数的区间分拆问题的三种算法,在不同的数据范围时表现各有优劣,需要结合具体的题目选择合适的算法.
例题 3:P4767 [IOI2000] 邮局 加强版 P6246 [IOI2000] 邮局 加强版 加强版
高速公路旁边有一些村庄.高速公路表示为整数轴,每个村庄的位置用单个整数坐标标识.没有两个在同样地方的村庄.两个位置之间的距离是其整数坐标差的绝对值.
邮局将建在一些,但不一定是所有的村庄中.为了建立邮局,应选择他们建造的位置,使每个村庄与其最近的邮局之间的距离总和最小.
你要编写一个程序,已知村庄的位置和邮局的数量,计算每个村庄和最近的邮局之间所有距离的最小可能的总和.
思路
每个村庄有其最近的邮局,那么每个邮局也有其管辖的村庄,易知这是一个区间.
考虑把这 \(n\) 个村庄分成 \(m\) 个区间,再在每个区间中决出一个邮局.
根据数学知识,对于区间 \([i,j]\),邮局应该建在第 \(\left\lfloor\dfrac{i+j}2\right\rfloor\) 个村庄处.使用前缀和容易算出 \(w(i,j)\).
问题转化为限制区间个数的区间分拆问题.可以证明,\(w\) 函数满足四边形不等式.直接应用上述优化方法即可.
实现 1,前文第二种优化,复杂度 \(O(n(n+m))\)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | |
实现 2,WQS 二分,复杂度 \(O(n\log n\log C)\)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | |
区间合并问题
另一类可以通过四边形不等式优化的动态规划问题是区间合并问题,即要将 \(n\) 个长度为一的区间 \([i,i]\) 两两合并起来,直到得到区间 \([1,n]\).每次合并 \([j,k]\) 和 \([k+1,i]\) 时都需要支付成本 \(w(j,i)\).问题要求找到成本最低的合并方式.对于此类问题,有如下 2D1D 状态转移方程:
其中,初始成本 \(f(i,i)=0\).暴力算法的总复杂度为 \(O(n^3)\),而当存在决策单调性时,可以优化至 \(O(n^2)\) 的算法复杂度.这一算法最早由 Knuth 在解决最优二叉搜索树问题时提出,并由姚储枫进一步研究总结,在国外称为 Knuth's optimization 或 Knuth-Yao speedup.
除了四边形不等式以外,区间合并问题的决策单调性还要求成本函数满足区间包含单调性.
-
区间包含单调性:如果对于任意 \(a \leq b \leq c \leq d\) 均成立
\[ w(b,c) \leq w(a,d), \]则称函数 \(w\) 对于区间包含关系具有单调性.
这实质是成本函数的一阶条件,即 \(w(j,i)\) 关于 \(j\) 递减,关于 \(i\) 递增.
引理 1
若 \(w\) 满足区间包含单调性和四边形不等式,则状态 \(f(j,i)\) 满足四边形不等式.
证明
不妨设 \(a \leq b \leq c \leq d\).下证 \(f(a,d) + f(b,c) \geq f(a,c) + f(b,d)\).考虑依 \(d-a\) 归纳.当 \(a=b\) 或 \(c=d\) 时,所求即一等式.对于一般的情形,根据 \(d'=\operatorname{opt}(a,d)\) 的位置分类讨论.
第一种情况,\(c \leq d'\) 或 \(d' < b\),即 \([b,c]\) 包含于 \([a,d']\) 或 \([d'+1,d]\) 之中.
不妨假设 \(c \leq d'\),另一种情形同理.此时有
这里,第一个不等式来自于归纳假设 \(f(a,c) + f(b,d') \leq f(a,d') + f(b,c)\),第二个不等式来自于区间包含单调性 \(w(b,d) \leq w(a,d)\),第三个不等式来自于最优性条件 \(f(b,d) \leq f(b,d') + f(d'+1,d) + w(b,d)\).
第二种情况,\(b \leq d' < c\),即 \(d'\) 位于 \([b,c]\) 之中.此时,考虑 \(c'=\operatorname{opt}(b,c)\) 的位置.
不妨假设 \(c' \leq d'\),即 \([b,c']\) 包含于 \([a,d']\) 之中,另一种情形同理.此时有
这里,第一个不等式来自于归纳假设 \(f(a,c') + f(b,d') \leq f(a,d') + f(b,c')\),第二个不等式来自于四边形不等式 \(w(a,c) + w(b,d) \leq w(a,d) + w(b,c)\),第三个不等式来自于 \(f(a,c)\) 和 \(f(b,d)\) 的最优性条件.
定理 4
若 \(w\) 满足区间包含单调性和四边形不等式,则问题 (3) 中最小最优决策 \(\operatorname{opt}(j,i)\) 满足
证明
引理 1 已经证得 \(f(j,i)\) 满足四边形不等式,所以目标函数 \(f(j,k) + f(k+1,i) + w(j,i)\) 对于给定 \(j\) 作为 \((k,i)\) 的函数满足四边形不等式,所以由定理 1 有,\(\operatorname{opt}(j,i-1) \leq \operatorname{opt}(j,i)\).注意,不同时含有 \((k,i)\) 的项并不影响四边形不等式成立.类似地,它对于给定 \(i\) 作为 \((k,j)\) 的函数也满足四边形不等式,所以 \(\operatorname{opt}(j,i) \leq \operatorname{opt}(j+1,i)\).即得所证.
利用这一结论,同样可以限制决策点 \(k\) 的搜索范围.在这里,正序遍历区间长度 \(i-j+1\),再遍历具有同样长度的所有区间 \([j,i]\),暴力搜索 \(\operatorname{opt}(j,i-1)\) 和 \(\operatorname{opt}(j+1,i)\) 之间的所有 \(k\) 求得最优解 \(f(j,i)\) 并记录最小最优决策 \(\operatorname{opt}(j,i)\).对于同样长度的所有区间,此算法中决策空间总长度是 \(O(n)\) 的,而可能的区间长度的数目同样是 \(O(n)\) 的,故而总的算法复杂度为 \(O(n^2)\) 的.
参考实现
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | |
满足四边形不等式的函数类
为了更方便地证明一个函数满足四边形不等式,我们有以下几条性质:
性质 1:若函数 \(w_1(j,i)\) 和 \(w_2(j,i)\) 均满足四边形不等式(或区间包含单调性),则对于任意 \(c_1,c_2\geq 0\),函数 \(c_1w_1+c_2w_2\) 也满足四边形不等式(或区间包含单调性).
性质 2:若存在函数 \(f(x)\) 和 \(g(x)\) 使得 \(w(j,i) = f(j)-g(i)\),则函数 \(w\) 满足四边形恒等式.当函数 \(f\) 和 \(g\) 单调增加时,函数 \(w\) 还满足区间包含单调性.
性质 3:设 \(h(x)\) 是一个单调增加的凸函数,若函数 \(w(j,i)\) 满足四边形不等式并且对区间包含关系具有单调性,则复合函数 \(h(w(j,i))\) 也满足四边形不等式和区间包含单调性.
性质 4:设 \(h(x)\) 是一个凸函数,若函数 \(w(j,i)\) 满足四边形恒等式并且对区间包含关系具有单调性,则复合函数 \(h(w(j,i))\) 也满足四边形不等式.
首先需要澄清一点,凸函数(Convex Function)的定义在国内教材中有分歧,此处的凸函数指的是下凸函数,即(可微时)一阶导数单调增加的函数.
证明
前两条性质根据定义很容易证明,下面证明第三条性质,性质四的证明过程类似.由于 \(h(x)\) 单调,\(h(w(j,i))\) 自然保持对区间包含的单调性.关键在于四边形不等式的证明.
为此,下面考虑 \(a \leq j \leq b \leq c \leq i \leq d\) 上的二阶混合差分.
这里,根据区间单调性,\(\Delta_iw(a,i) := w(a,d) - w(a,c) \geq 0\) 和 \(\Delta_jw(j,c) := w(b,c) - w(a,c) \leq 0\).由于 \(h(x)\) 具有凸性,对于 \(t_1,t_2\geq 0\) 成立 \(h(x + t_1 - t_2) - h(x + t_1) \leq h(x - t_2) - h(x)\),所以后两行必然非正.同时,由于四边形不等式,\(w(b,d) \leq w(a,c) + \Delta_jw(j,c) + \Delta_iw(a,i) = w(b,c) + w(a,d) - w(a,c)\),故而,第一行的差在 \(h(x)\) 单调增加的情况下必然也非正.所以,总的二阶混合差分非正.此即四边形不等式.
这一证明实际是如下导数证明的离散版本.
这在 \(h' \geq 0\),\(h'' \geq 0\),\(w_x \leq 0\),\(w_y \geq 0\) 以及 \(w_{xy} \leq 0\) 的条件下显然成立.其中,区间包含单调性给出了 \(w\) 的一阶条件,而四边形不等式给出了其二阶条件.
习题
- Codeforces - Ciel and Gondolas(Be careful with I/O!)
- SPOJ - LARMY
- Codechef - CHEFAOR
- Hackerrank - Guardians of the Lunatics
- ACM ICPC World Finals 2017 - Money
参考资料与注释
- Quora Answer by Michael Levin
- Video Tutorial by "Sothe" the Algorithm Wolf
- Divide and Conquer DP
- Knuth's Optimization
- Quadrangle Inequality Properties
- 王钦石《浅析一类二分方法》
- 簡易版 LARSCH Algorithm by noshi91
- 四边形不等式和决策单调性 by b6e0_ - 洛谷专栏
- 在线决策单调性的丐版 LARSCH 算法 by Register_int - 洛谷专栏
Last updated on this page:, Update history
Found an error? Want to help improve? Edit this page on GitHub!
Contributors to this page:Marcythm, zyf0726, hsfzLZH1, MingqiHuang, Ir1d, greyqz, billchenchina, Chrogeek, StudyingFather, NFLSCode, c-forrest
All content on this page is provided under the terms of the CC BY-SA 4.0 and SATA license, additional terms may apply