跳转至

Slope Trick 优化

引入

对于一类二维 DP 问题,如果它的价值函数 f(i,x) 对于每个固定的 i 都是 x 的凸函数,那么将函数 f(i,) 整体视为 i 处的状态,并维护它的差分(或斜率)

Δf(i,x)=f(i,x+1)f(i,x)

而非函数本身,往往能够起到优化转移的效果。这种优化 DP 的思想,就称为 Slope Trick。

「斜率」

因为大多数题目中涉及的函数都只在整点处取值,所以称它为差分和斜率没有本质区别,本文按照 Slope Trick 这个名词统一称呼它为斜率。

具体题目中,斜率的维护方式可能各不相同。如果斜率的取值范围较窄,维护斜率变化的点(即拐点)更为方便;而如果函数定义域较窄,维护斜率序列本身可能更为方便。更复杂的情形,可能需要同时维护每段斜率的大小和该段的长度。无论具体维护方式是什么,这类问题的本质都是利用状态转移中斜率序列变化较少这一点简化转移。因此,它们都可以称作 Slope Trick。

凸函数

在讨论具体的题目之前,有必要首先了解一下凸函数的基本性质,以及在对凸函数进行各种变换时,它的斜率会如何变化。

实轴上的凸函数

凸函数较为一般的定义是在 R 上给出的。

R 上的凸函数

如果函数 f:RR{±} 对于所有 x,yRα(0,1) 都满足

f(αx+(1α)y)αf(x)+(1α)f(y),

就称函数 f凸函数(convex function),其中 ± 的运算法则规定为 ± 乘以任何正实数或是加上任何实数都等于其自身,且对于任何实数 xR 都有 <x<+

当然,如果不等号换作 ,就相应地称它为凹函数1。因为对于凹函数 f,总有 f 为凸函数,所以本节只考虑凸函数。

本文只考虑正常凸函数

为了避免讨论 的取值和额外的复杂分析,本文在讨论凸函数相关概念时,总是默认函数不会取到 ,且不总是 +。这样的凸函数称为 正常凸函数(proper convex function)。这对于理解算法竞赛涉及的内容已经足够。

当然,函数 f 往往并不会对所有实数都有定义。如果函数 f 的定义域仅是 R 的子集,那么可以将它拓展为 R 上的函数:

f~(x)={f(x),xdomf,+,xdomf.

此时,称 f 是凸函数,当且仅当相应的 f~ 满足上述凸函数的定义。因此,如果没有特别指出,本文提到的凸函数的定义域均是实数集 R。显然,凸函数 f 只能在一个区间(即 R 的凸子集)上取得有限值。

简单例子

常见的凸函数的例子包括:

  1. 常数函数:f(x)=c,其中 cR
  2. 一次函数:f(x)=kx+b,其中 k,bRk0
  3. 绝对值函数:f(x)=|xa|,其中 aR
  4. 任何凸函数限制在某个区间上的结果,例如 0[a,b](x)(在凸分析的语境下也称作 [a,b] 的指示函数)。

当然,可以通过下文提到的保持凸性的变换组合出更为复杂的凸函数。

离散点集上的凸函数

算法竞赛中,很多函数仅在部分整数值处有定义。它们在一般情况下并不是(上文定义的)凸函数,因为它们的定义域不再是凸集。为了处理这种情形,需要单独定义离散点集上的函数的凸性。简单来说,需要首先对函数做线性插值,将其定义域拓展到区间,再判断它的凸性。

离散点集上的凸函数

SR 为离散点集,即对任意闭区间 [a,b]S[a,b] 都是有限集。对于函数 f:SR{±},可以定义函数 f~:RR{±} 使得:

  • xS 时,f~(x)=f(x)
  • x(infS,supS)S 时,设 s=max{sS:sx}s+=min{sS:sx},则

    f~(x)=s+xs+sf(s)+xss+sf(s+),
  • x[infS,supS] 时,f~(x)=+

那么,如果 f~(x)R 上的凸函数,就称 f(x)S 上的 凸函数

因为 R 上的凸函数处理起来更为方便,所以本文在提及凸函数时,若非特别说明,指的都是 R 上的凸函数。如果本文中某个函数仅给出了部分整数处的取值,那么它在其他实数处的取值应由定义中的 f~ 确定,也就相当于直接讨论对应的分段线性函数 f~

整数集 Z 上的凸函数有一个更为直观的等价定义:

Z 上的凸函数的等价定义

函数 f:ZR{±} 是凸的,当且仅当

f(x)f(x1)f(x+1)f(x)

对于所有 xZ 都成立。

证明

这个命题是凸函数的斜率刻画的简单推论。

如果 fZ 上的凸函数,那么根据斜率弱增,有

Δf(x1,x)Δf(x1,x+1)Δf(x,x+1).

这就是上述条件。

反过来,如果上述条件成立,那么对于任何 x1<x2,都有

Δf(x1,x2)=1x2x1i=x1x21(f(i)f(i1)).

这相当于对所有满足 x1i<x2 的差分的算术平均值。如果 x2 增加一,就相当于插入一项更大的差分;如果 x1 增加一,就相当于移除一项最小的差分。这两个操作都会使得平均值上升。这就说明斜率 Δf(x1,x2) 弱增,即 fZ 上的凸函数。

也就是说,只要斜率(差分)单调不减,这个序列就可以看作是 Z 上的凸函数。

凸函数的两种刻画

其实,用斜率刻画凸函数的方式也可以推广到一般情况。

凸函数的斜率刻画

SR 或它的离散子集,则函数 f:SR{±} 为凸函数,当且仅当斜率

Δf(x1,x2)=f(x2)f(x1)x2x1

对于任何 x1,x2Sx1<x2 都是 x1x2 的弱增函数。

证明

对于 R 上的函数 f(x) 以及 x1<x2,对于 α(0,1),令 x3=αx1+(1α)x2,那么

Δf(x1,x3)Δf(x1,x2)Δf(x3,x2)

就等价于

f(x3)f(x1)1αf(x2)f(x1)f(x2)f(x3)α.

这两侧的不等式都等价于 f(x3)αf(x1)+(1α)f(x2),即函数 f(x) 的凸性。

对于 R 的离散子集 S 上的函数 f(x),斜率弱增这一条件的必要性可以由 f~(x) 的凸性推导出来。现在要证明它的充分性,为此只要证明 Δf~(x1,x2) 也是弱增的。设 S={si}si 关于 i 严格递增,并设 si1x1si1+1si2x2si2+1,自然有 i1i2。令 Δi=Δf(si,si+1),那么,可以证明 Δi1Δf~(x1,x2)Δi2

这分两种情形。如果 i1=i2,那么 Δi1=Δf~(x1,x2)=Δi2,该不等式显然成立。否则,有

Δf~(x1,x2)=1x2x1((si1+1x1)Δi1+(x2si2)Δi2+j=i1+1i21(sj+1sj)Δj).

根据 S 上的斜率递增可知,Δi 关于 i 递增,所以,Δi1Δf~(x1,x2)Δi2

利用这个结论,对于 x1<x2α(0,1),令 x3=αx1+(1α)x2,并取 i3 使得 si3x3si3+1 成立,则有

Δf~(x1,x3)Δi3Δf~(x3,x2).

代入 x3 的表达式,就得到 f~(x) 的凸性。

斜率单调不减,可以看作是凸函数的等价定义。正因为凸函数的斜率具有单调性,在维护斜率时,通常需要选择 堆(优先队列)平衡树 等数据结构。

本文还会用到凸函数的另一种等价刻画。对于函数 f:RR{±},可以考察平面内函数图像上方的区域,即

epif={(x,y)R2:yf(x)}.

这个区域也称为函数 f上境图(epigraph)。函数的凸性,等价于它的上境图的凸性:

凸函数的上境图刻画

函数 f:RR{±} 是凸函数,当且仅当 epifR2 内的凸集。

证明

如果 f 是凸函数,那么对于 (x1,y1),(x2,y2)epif 和任意 α(0,1),有

αy1+(1α)y2αf(x1)+(1α)f(x2)f(αx1+(1α)x2).

所以,α(x1,y1)+(1α)(x2,y2)epif

反过来,如果 epif 是凸集,那么对于任意 x1<x2 以及 α(0,1),有

α(x1,f(x1))+(1α)(x2,f(x2))epif.

这就等价于 αf(x1)+(1α)f(x2)f(αx1+(1α)x2),即 f 的凸性。

稍后会看到,利用上境图,可以将凸函数的卷积下确界与凸集的 Minkowski 和联系起来。

凸函数的变换

紧接着,本文介绍一些 Slope Trick 中经常遇见的保持凸性的变换。

非负线性组合

对于凸函数 fg 以及非负实数 α,β0,函数 αf+βg 也是凸函数。而且,

Δ(αf+βg)=αΔf+βΔg.

因此,如果维护了凸函数 fg 的斜率,要得到它们的非负线性组合 αf+βg 的斜率,只需要逐段计算即可。

在维护斜率的问题中,往往其中一个函数的形式比较简单,此时可以通过懒标记的方式降低修改复杂度。在维护拐点的问题中,要计算 f+g 的斜率拐点,只需要将 fg 的斜率拐点合并即可。

卷积下确界(Minkowski 和)

凸函数的另一种常见操作是卷积下确界。对于函数 fg,函数

h(x)=infyRf(y)+g(xy)

称为 fg卷积下确界2(infimal convolution)。如果 fg 都是凸函数,它们的卷积下确界也是凸函数。

对图示的解释

如图所示,要求 fg 的卷积下确界 h,可以将 f 的图像(第三个图的红色虚线)上的每一个点都视作原点,在相应的坐标系内画出 g 的图像(第三个图中的蓝色虚线)。当坐标系原点沿着 f 的图像移动时,g 的图像(上境图)移动的轨迹轮廓(即下凸壳),就是 h 的图像。可以看出,h 的每一个斜率段,都要么是 f 的斜率段,要么是 g 的斜率段:只是重新按照斜率大小排序了。这个过程中,fg 的角色可以互换,即让 f 的图像沿着 g 的图像移动,得到的结果是一致的。

几何直观上,epih 就是 epifepigMinkowski 和。如果 fg 都是分段线性函数,那么 h 同样是分段线性函数,且它的斜率段可以看作是 fg 的斜率段合并(再排序)的结果。

证明

f,g 都是凸函数,h 是它们的卷积下确界。设 x1<x2,且 α(0,1)。根据卷积下确界的定义,对任意 ε>0,存在 yi,ziR 使得 yi+zi=xi

h(xi)+ε>f(yi)+g(zi).

故而,结合 f,g 的凸性及 h 的定义,有

αh(x1)+(1α)h(x2)+ε>αf(y1)+(1α)f(y2)+αg(z1)+(1α)g(z2)f(αy1+(1α)y2)+g(αz1+(1α)z2)h(αx1+(1α)x2).

因为 ε>0 是任意选取的,所以

αh(x1)+(1α)h(x2)h(αx1+(1α)x2).

这就得到 h 的凸性。

然后,对于几何直观,严格地说,只能证明如下结论:

epif+epigepihcl(epif+epig).

其中,cl 表示闭包。

对于任何 (x,y)epif+epig,都存在 (x1,y1)epif(x2,y2)epig,使得 x=x1+x2

y=y1+y2f(x1)+g(x2)h(x1+x2)=h(x).

故而,(x,y)epih。这说明 epif+epigepih

反过来,对于任何 (x,y)epih,有 yh(x)。根据 h 的定义,对任何 ε>0,都存在 x1+x2=x 使得

y+ε>f(x1)+g(x2).

y1=f(x1)y2=g(x2),就有 y+ε>y1+y2。这说明,对于任何 ε>0,都有 (x1,y1)+(x2,y2)epif+epig 位于点 (x,y) 与点 (x,y+ε) 的连线上。取 ε0,就有 epihcl(epif+epig)

所以,epif+epig=epih 当且仅当它是闭凸集。一个使其满足的条件是,fg 都是正常凸函数且 下半连续。对于算法竞赛的应用来说,这已经足够了,比如分段线性函数总是满足这些条件的。

在实际问题中,如果 fg 其中一个的斜率段数较少,可以直接将较少的斜率段插入到较多的斜率段中;否则,可能需要利用 启发式合并可并堆 等方法,降低合并的整体复杂度,或者根据具体问题寻找相应的处理方式。

最值操作

两个凸函数的最大值仍然是凸函数,但是,两个凸函数的最小值未必仍然是凸函数。

很多常见的最小值操作可以转化为卷积下确界:

例子
  • f(x)=miny[x+a,x+b]g(y) 仍然是凸函数,因为它可以看作是卷积下确界:

    f(x)=minyRg(y)+0[b,a](xy).
  • f(x)=min{g(xai)+bi}Z 上的凸函数,只要 g(x)Z 上的凸函数,且在有限集合 {ai}Z 上定义的函数 h:aibi 也是该离散集合上的凸函数。这是因为延拓之后的函数 f~(x) 可以看作是卷积下确界:

    f~(x)=minyRh~(y)+g~(xy).

    因此,延拓之前的函数 f(x) 也是凸函数。

但并不是所有的最小值操作都保持凸性。

反例

g(x) 是凸函数,函数 f(x)=min{g(x1)+kx,g(x)} 并不一定是凸函数。

在一些特殊的问题中,尽管动态规划的转移方程可以写作两个凸函数的最小值的形式,且难以转化为卷积下确界的形式,但是价值函数依然能够保持凸性。在实际处理时,通常需要结合打表和猜测找到这类问题的合理的斜率转移方式。

了解了凸函数及其常见变换后,就可以通过具体的问题理解 Slope Trick 优化 DP 的方法。本文的例题大致分为维护拐点和维护斜率两组,用于理解这两种维护方式的常见操作和实施细节。但是,正如前文所强调的那样,维护方式并不是 Slope Trick 的本质,应当根据具体的问题需要选取合适的斜率段维护方式。

维护拐点

这类问题通常出现在需要最小化若干个绝对值的和式的问题中。因为这类问题中,价值函数的斜率的绝对值并不大,因此维护斜率变化的拐点更为方便。

维护拐点是指维护分段线性函数中,斜率发生变化的点。相当于对于每个斜率为 ki 的斜率段 [li,ri],只维护其端点信息,而斜率本身不需要格外维护;因此,这类问题斜率每次发生变化时,都应当只变化一个固定的量。比如,如果维护了拐点集 ξsξ1ξ1ξt,就相当于说:区间 [ξ1,ξ1] 内斜率为 0;向左每经过一个拐点,斜率减少一;向右每经过一个拐点,斜率增加一;故而,区间 [ξ2,ξ3] 内,斜率就是 2,区间 [ξ3,ξ2] 内,斜率就是 2,诸如此类。用形式语言表示,函数可以利用斜率拐点写作

f(x)=f(ξ1)+i=s1max{ξix,0}+i=1max{xξi,0}.

它的最小值就是 f(ξ1)=f(ξ1),且可以在区间 [ξ1,ξ1] 内任意位置取到。

例题:最小成本递增序列

[BalticOI 2004] Sequence 数字序列

给定长度为 n 的序列 {ai},求严格递增序列 {bi} 使得 i|aibi| 最小,输出最小值和任意一种最优方案 {bi}

解答

首先,{bi} 严格递增,等价于 {bi}={bii} 弱增。因此,可以对 {ai}={aii} 求出差值最小的弱增序列 {bi}={bii} 再恢复成序列 {bi} 即可。

考虑朴素 DP 解法。设 fi(x) 是已经选取了序列 {bi} 中前 i 个数字,且第 i 个数字不超过 x 时,已经选取的数字与 {ai} 的前 i 个数字的最小差值:

fi(x)=minj=1i|ajbj| s.t. b1b2bix.

容易得到状态转移方程为

fi(x)=minyxfi1(y)+|aiy|.

初始状态为 f0(x)0,最后要求的就是 minxfn(x)。利用前文提到的凸函数的变换,从 fi1(x)fi(x),需要经过两步变换:

  1. 首先,加上 |aix|,这相当于对区间 (,ai] 内的所有斜率段都增加 1,对区间 [ai,+) 内的所有斜率段都增加 1
  2. 对得到的函数取最小值,将 g(x)=fi1(x)+|aix| 变为 fi(x)=minyxg(y)。根据前文分析,这相当于对 g(x)0[0,+) 做卷积下确界。因为后者的斜率段只有一段,斜率为 0 且向右延伸至无限长,将其插入 g(x) 的斜率段中,相当于删除其中所有正斜率段。

明晰了这些操作后,已经可以直接用平衡树维护所有斜率段了,但代码较复杂。注意到问题中斜率每次变化至多 1,故而所有斜率段的绝对值都不超过 n。不直接维护斜率段,转而直接维护斜率拐点更为方便。

f1(x) 的拐点集为 ξkξ1ξ1ξ。那么,上面的两步操作分别对应:

  1. 增加一个负斜率段的拐点 ai 和一个正斜率段的拐点 ai
  2. 弹出所有正斜率段的拐点 ξ1,,ξ

实际维护时,因为每次操作结束后都没有正斜率段的拐点,即斜率拐点具有形式 ξkξ1,而且操作总发生在正负斜率段交界处,所以直接维护一个最大堆存储所有拐点即可。两步操作分别对应:

  1. 插入两次 ai
  2. 弹出堆顶。

当然,每次结束后都需要维护当前函数的最小值。因为操作结束后,没有正斜率段,函数最小值就是它在最大堆堆顶处的取值。设每次操作之前堆顶为 ξ1,最小值为 fi1(ξ1)。因为弹出的堆顶是正斜率段的最小拐点,函数的最小值就等于该处函数的取值,所以直接计算弹出前堆顶处函数的取值即可,亦即

fi1(max{ai,ξ1})+|max{ai,ξ1}ai|=fi1(ξ1)+max{0,ξ1ai}.

其中,第一项相等是因为 fi1(x) 没有正斜率段。因此,每次只需要在最小值上不断累加 max{0,ξ1ai} 即可。

本题还要求输出一种最优方案。因为最后操作结束时,最优解就是堆顶,所以 bn 的取值可以直接确定。如果已经知道了第 i 个最优解 bi,要求解 fi1(x) 满足 xb 的最优解,只需要注意到因为 fi1(x) 是凸的,所以越接近它的全局最小值点,解就越优,故而只要记录 fi1(x) 的全局最小值点,并将它与 bi 取最小值,就可以得到最优的 bi1

时间复杂度为 O(nlogn)

 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
#include <iostream>
#include <queue>
#include <vector>

int main() {
  int n;
  std::cin >> n;
  std::vector<int> a(n), b(n);
  for (int& x : a) std::cin >> x;
  for (int i = 0; i < n; ++i) a[i] -= i;
  long long res = 0;
  std::priority_queue<int> max_heap;
  for (int i = 0; i < n; ++i) {
    max_heap.emplace(a[i]);
    max_heap.emplace(a[i]);
    res += max_heap.top() - a[i];
    max_heap.pop();
    b[i] = max_heap.top();
  }
  std::cout << res << '\n';
  for (int i = n - 2; i >= 0; --i) b[i] = std::min(b[i], b[i + 1]);
  for (int i = 0; i < n; ++i)
    std::cout << (b[i] + i) << (i == n - 1 ? '\n' : ' ');
  return 0;
}

模板题:

例题:转移带限制的情形

[NOISG 2018 Finals] Safety

给定长度为 n 的序列 {ai},求序列 {bi} 使其满足 |bibi1|h 对所有 1<in 都成立,并使得 i|aibi| 最小,输出最小值。

解答

内容大致与上一个题目相仿,只是序列 {bi} 的限制发生了变化。同样地,设 fi(x) 为第 i 个数字取 x 时,前 i 个数字的差值的最小值:

fi(x)=minj=1i|ajbj| s.t. |bj1bj|h,1<ji, bi=x.

由此,有状态转移方程为

fi(x)=|aix|+min|yx|hfi1(y).

起始条件为 f0(x)0。最后要求的仍然是 minxfn(x)

状态转移拆解为对凸函数的操作,分两步:

  1. 首先对 fi1(x) 取最值,变为 min|yx|hfi1(y),这相当于 fi1(x)0[h,h](x) 的卷积下确界;
  2. 再将得到的函数与 |aix| 相加。

同样因为斜率每次只变化一,可以考虑维护拐点。这样,这两步操作就可以描述为:

  1. 将所有负斜率段向左移动 h,将所有正斜率段向右移动 h
  2. 插入两次 ai

显然,对于本题,将正负斜率段分别维护较为方便。因为操作主要集中在零斜率段附近,因此考虑使用 对顶堆,即分别用最大堆和最小堆维护负斜率段和正斜率段的拐点。拐点的整体平移操作用懒标记完成。因为第二步操作需要分别对两个堆插入一个 ai,而且,插入完成后,未必最大堆的堆顶仍然小于等于最小堆的堆顶。此时,交换两堆顶,直到堆顶的大小关系得到满足即可。

最后,考虑操作过程中如何更新最小值。因为第一步平移操作并不会改变最小值,所以只要考虑交换堆顶的操作即可。设 ξ1>ξ1,将堆顶 ξ1ξ1 交换时,函数由

max{0,xξ1}+max{0,xξ1}

变为

max{0,xξ1}+max{0,xξ1}.

过程中,函数形状不变,只是向下平移了 |ξ1ξ1|。因此,要使得交换堆顶前后函数保持不变,只需要将最小值累加 |ξ1ξ1| 即可。

算法的时间复杂度仍为 O(nlogn),因为每次添加元素后,交换堆顶的操作至多执行一次。

 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
#include <iostream>
#include <queue>
#include <vector>

int main() {
  int n;
  long long h;
  std::cin >> n >> h;
  std::priority_queue<long long> max_heap;
  std::priority_queue<long long, std::vector<long long>, std::greater<>>
      min_heap;
  long long lt = 0, rt = 0;
  long long res = 0;
  for (; n; --n) {
    long long x;
    std::cin >> x;
    lt += h;
    rt += h;
    max_heap.emplace(x + lt);
    min_heap.emplace(x - rt);
    auto l = max_heap.top() - lt;
    auto r = min_heap.top() + rt;
    while (l > r) {
      max_heap.pop();
      min_heap.pop();
      res += l - r;
      max_heap.emplace(r + lt);
      min_heap.emplace(l - rt);
      l = max_heap.top() - lt;
      r = min_heap.top() + rt;
    }
  }
  std::cout << res << std::endl;
  return 0;
}

模板题:

维护斜率

还有一些问题,维护斜率更为方便。这类问题通常也可以使用 反悔贪心 或模拟费用流的思想解决。费用流模型中,最小费用往往是流量的凸函数,这就为使用 Slope Trick 提供了基础。

例题:股票交易问题

Codeforces 865 D. Buy Low Sell High

给定 n 天股票价格序列 {pi}(均为正数),初始持股为 0,每天可买入一股、卖出一股或不交易,求 n 天后最大利润。

解答

首先考虑朴素 DP 解法。设 fi(x) 为第 i 天结束时持有股票数量为 x0 的最大利润,则

fi(x)=max{fi1(x1)pi,fi1(x),fi1(x+1)+pi}.

初始状态为 f0(0)=0,且对所有 x0,有 f0(x)=。问题的答案就是 fn(0)

fi1(x)fi(x) 需要经过两步变换:

  1. fi1(x) 与函数

    hi(x)={pi,x=1,0,x=0,pi,x=1

    对应的分段线性函数 h~(x)(显然是凹函数)做卷积上确界;

  2. 因为这样会导致函数在区间 [1,0) 内具有有限值,这与 x0 的要求矛盾,故而需要截取函数在 [0,+) 内的部分。

将它们转化为斜率段的变化,就是如下两步:

  1. 插入长度为 2、斜率为 pi 的斜率段;
  2. 删除斜率有限的斜率段中,斜率最大且长度为 1 的一段。

因为斜率段的长度总是自然数,所以不妨维护若干个长度为一的斜率段,从而只需要记录每段的斜率即可。因为只需要插入和访问最大值操作,所以只需要一个最大堆。操作分两步:

  1. 插入两次 pi
  2. 弹出堆顶。

还需要维护 fi(0) 的值。因为第一步操作得到的函数在 x=1 处的取值就是 fi1(0)+pi,所以它在 x=0 处的取值就是该值加上马上要弹出的堆顶——它就是函数在区间 [1,0] 上的斜率。因为截断不改变函数在 x=0 处的取值,所以这就是 fi(0)

对比该算法实现与上文 最小成本递增序列 的代码可知,该算法等价于求将股票价格变为弱递减序列的最小成本。

时间复杂度为 O(nlogn)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
#include <iostream>
#include <queue>
#include <vector>

int main() {
  int n;
  std::cin >> n;
  std::vector<int> a(n);
  for (int& x : a) std::cin >> x;
  long long res = 0;
  std::priority_queue<int> max_heap;
  for (int x : a) {
    max_heap.emplace(-x);
    max_heap.emplace(-x);
    res += x + max_heap.top();
    max_heap.pop();
  }
  std::cout << res << '\n';
  return 0;
}

模板题:

例题:搬运土石问题

[USACO16OPEN] Landscaping P

给定长度为 n 的序列 {ai}{bi},分别表示第 i 个花园已经有的泥土数量和需要的泥土数量(不能多也不能少)。购买一单位泥土放入任意花园价格为 X,从任意花园运走一单位泥土价格为 Y,从花园 i 向花园 j 运送一单位泥土价格为 Z|ij|。求满足所有花园需求的最小成本。(ai,bi10

解答

考虑朴素 DP 解法。设 fi(x) 为满足前 i 个花园需求且净剩余 x 单位泥土运到后面的花园时的最小代价。如果 x<0,就相当于净亏空 |x| 单位泥土需要从后面的花园运送过来。那么,可以写成状态转移方程为

fi(x)=minyRfi1(y)+|y|Z+h((xy)+(biai)).

其中,函数 h(δ) 表示当前花园的泥土净购买量为 δ 时的成本,即

h(δ)=max{0,δ}X+max{0,δ}Y=max{δX,δY}.

它显然是凸函数。该状态转移方程的含义为

  • 之前 i1 个花园净剩余泥土数量为 y 时,最小成本为 fi1(y)
  • 将净剩余(亏空)的泥土数量在 ii1 之间运送的成本为 |y|Z
  • 通过买卖,将第 i 个花园的泥土数量从 ai 调整为 bi,并将净剩余泥土数量从 y 调整到 x,最小成本为 h((xy)+(biai))

初始状态为 f0(0)=0,且对所有 x0,有 f0(x)=+。问题的答案就是 fn(0)

将函数 fi1(x) 变换为 fi(x) 可以分为三步:

  1. 首先,加上 |x|Z,得到 fi1(x)+|x|Z
  2. 然后,与 h(x) 做卷积下确界,得到 minyRfi1(y)+|y|Z+h(xy)
  3. 最后,将函数向左平移 (biai) 个单位。

转化为对斜率段的操作,同样分为三步:

  1. 将原点左侧斜率段全体加上 Z,将原点右侧斜率段全体加上 Z
  2. 将所有小于 Y 的斜率段全部替换为 Y,将所有大于 X 的斜率段全部替换为 X
  3. 将所有斜率段向左平移 (biai) 个单位。

原题中 aibi 很小,因此只需要维护若干个长度为 1 的斜率段即可。虽然斜率段有无穷多个,但是有上界 X 和下界 Y,且严格位于两者之间的斜率段数目并不多。因为不涉及插入操作,所以可以用两个栈维护原点两侧的斜率段,区间加和区间最值操作全部打懒标记完成。上述三步操作分别对应:

  1. 对左右两个栈分别打懒标记,左侧加 Z,右侧加 Z
  2. 每次栈内弹出元素时,都对 Y 取最大值,对 X 取最小值。如果左栈为空,则弹出 Y。如果右栈为空,则弹出 X
  3. 将左栈顶部的 (biai) 个元素弹出,插入右栈;当然,biai<0 时,就反过来。

在交换栈顶时,更新答案,向左移动就减去当前斜率,向右移动就加上当前斜率。

算法复杂度为 O(nmax{ai,bi})

 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
#include <iostream>
#include <stack>

int main() {
  int n;
  long long x, y, z;
  std::cin >> n >> x >> y >> z;
  std::stack<long long> neg, pos;
  long long lt = 0, rt = 0;
  long long res = 0;
  for (; n; --n) {
    int a, b;
    std::cin >> a >> b;
    lt -= z;
    rt += z;
    for (; b < a; ++b) {
      auto l = -y;
      if (!neg.empty()) {
        l = std::max(l, neg.top() + lt);
        neg.pop();
      }
      pos.emplace(l - rt);
      res -= l;
    }
    for (; b > a; --b) {
      auto r = x;
      if (!pos.empty()) {
        r = std::min(r, pos.top() + rt);
        pos.pop();
      }
      neg.emplace(r - lt);
      res += r;
    }
  }
  std::cout << res << std::endl;
  return 0;
}

模板题:

习题

本文的最后,提供一些各类算法竞赛中出现过的且可以使用 Slope Trick 解决的问题,以供练习。

参考文献与注释


  1. 不同教材对于凸函数的称呼可能不同。 

  2. 也常称为 min 卷积、inf 卷积或者 (min,+) 卷积。