【学习笔记】DP 优化 4:决策单调性
概述
根据贡献函数的特殊性质,得出最优决策点的单调性,可以大幅优化 DP 的复杂度。
四边形不等式
下面默认问题要求最小化。
设贡献函数 \(w(l,r)\) 为整数域上二元函数,若对于任意 \(l_1\le l_2\le r_1\le r_2\) 满足:
则称 \(w(l,r)\) 满足四边形不等式,记为“交叉优于包含”。
定理:若对于任意 \(l\le r\),满足 \(w(l,r)+w(l+1,r+1)\le w(l,r+1)+w(l+1,r)\),则 \(w(l,r)\) 满足四边形不等式。
先证明对于任意 \(l_1\le l_2\le r\),有 \(w(l_1,r)+w(l_2,r+1)\le w(l_1,r+1)+w(l_2,r)\),对 \(i=l_2-l_1\) 归纳,\(i=0\) 时显然成立。
根据假设,有:
根据条件,有:
两式相加就有:
之后证明对于满足四边形不等式,对 \(j=r_2-r_1\) 归纳,\(j=0\) 时显然成立:
根据假设,有:
根据上面证明的结论,有:
两式相加就有:
设转移方程:
设 \(p_i\) 为 \(f_i\) 的最优决策点,若贡献函数 \(w(j,i)\) 满足四边形不等式,则对于任意 \(i_1<i_2\),\(p_{i_1}\le p_{i_2}\)。
考虑反证法,若 \(p_{i_1}>p_{i_2}\),则 \(w(p_{i_2},i_2)\le w(p_{i_1},i_2)\) 且 \(w(p_{i_1},i_1)<w(p_{i_2},i_1)\),移项就是 \(w(p_{i_2},i_2)-w(p_{i_1},i_2)\le 0<w(p_{i_2},i_1)-w(p_{i_1},i_1)\),即 \(w(p_{i_1},i_1)+w(p_{i_2},i_2)<w(p_{i_2},i_1)+w(p_{i_1},i_2)\),与四边形不等式不符。
优化 2D/1D 区间 DP:记录决策点
若贡献函数 \(w(l,r)\) 满足对于任意 \(l_1\le l_2\le r_1\le r_2\) 满足:
则称 \(w(l,r)\) 满足区间包含单调性,这里的单调性同题目的要求。
设 2D/1D 区间 DP 转移方程:
定理:若贡献函数 \(w(l,r)\) 满足四边形不等式和区间包含单调性,那么 \(f_{l,r}\) 满足四边形不等式,即对于任意 \(l_1\le l_2\le r_1\le r_2\) 满足 \(f_{l_1,r_1}+f_{l_2,r_2}\le f_{l_1,r_2}+f_{l_2,r_1}\)。
对 \(r_2-l_1\) 归纳,讨论 \(f_{l_1,r_2}\) 的最优决策点 \(u\) 的位置。
若 \(u\ge r_1\) 或 \(u<l_2\),则 \([l_2,r_1]\) 被 \([l_1,u]\) 或 \([u+1,r_2]\) 包含,以 \(u\ge r_1\) 为例,有:
\(u<l_2\) 的情况同理。
若 \(r_1<u\le l_2\),设 \(f_{l_2,r_1}\) 的最优决策点 \(v\),以 \(v\le u\) 为例,有:
定理:设 \(p_{l,r}\) 为 \(f_{l,r}\) 的最优决策点,若 \(f_{l,r}\) 满足四边形不等式,则 \(p_{l,r-1}\le p_{l,r}\le p_{l+1,r}\)。
记 \(u=p_{l,r},v_1=p_{l,r-1},v_2=p_{l+1,r}\),考虑反证法:
若 \(u<v_1\),则 \(u+1<v_1+1\le r-1< r\),由四边形不等式知:
同时有:
两式相加就有:
与 \(v_1\) 是 \(f_{l,r-1}\) 的最优决策点不符。
若 \(u>v_2\),则 \(l<l+1\le v_2<u\),由四边形不等式知:
同时有:
两式相加就有:
与 \(v_2\) 是 \(f_{l+1,r}\) 的最优决策点不符。
有了上面的定理,在区间 DP 过程中,记录下每个状态的最优决策点,枚举的中间点 \(k\) 的范围就可以缩小,发现对于长度相等的区间,\(k\) 的范围是若干只在端点处相交的区间,那么每一层的复杂度是 \(O(n)\),总时间复杂度就是 \(O(n^2)\)。
满足四边形不等式或区间包含单调性的函数类
一些比较复杂的函数可以通过以下性质判断是否满足四边形不等式或区间包含单调性。
性质 1:若函数 \(w_1(l,r)\) 与 \(w_2(l,r)\) 均满足四边形不等式(或区间包含单调性),则对于任意 \(c_1,c_2\ge 0\),函数 \((c_1w_1+c_2+w_2)(l,r)\) 页满足四边形不等式(或区间包含单调性)。
性质 2:若能将 \(w(l,r)\) 表示为 \(f(l)-g(r)\),则函数 \(w(l,r)\) 满足四边形恒等式,若 \(f,g\) 均单调增加,函数 \(w(l,r)\) 还满足区间包含单调性。
性质 3:设 \(h(x)\) 是一单调增加的下凸函数,若函数 \(w(l,r)\) 满足四边形不等式和区间包含单调性,则复合函数 \(h(w(l,r))\) 也满足四边形不等式和区间包含单调性。
性质 4:设 \(h(x)\) 是一下凸函数,若函数 \(w(l,r)\) 满足四边形恒等式和区间包含单调性,则复合函数 \(h(w(l,r))\) 也满足四边形不等式。
证明略。
优化离线决策单调性:决策单调性分治
在四边形不等式得到的决策单调性中,若转移方程是 \(g+w\to f\) 的转移,即转移是由一个完全已知的函数或是 DP 数组的上一层得来的,将这种决策单调性视作“离线的”。
使用决策单调性分治优化 DP,分治函数 \(solve(l,r,L,R)\),表示当前的转移区间 \([l,r]\),决策点的区间 \([L,R]\),每次取 \(mid=\left\lfloor\frac{l+r}{2}\right\rfloor\),暴力遍历 \([L,R]\),找到最优决策点 \(p\),递归处理 \(solve(l,mid-1,L,p)\) 与 \(solve(mid+1,r,p,R)\)。每次的复杂度 \(O(n\log n)\)。
在分治过程中,存在贡献函数不是一个表达式的情况,即无法独立地快速求出每个贡献函数值。如果支持快速移动指针,类似莫队的处理,维护一个当前的区间,每次查询贡献函数时直接暴力移动过去即可。
复杂度证明可以对左右指针分别考虑,左指针在决策区间的分治树上,右指针在位置区间的分治树上,不难发现每次移动次数都是区间长度的常数倍。
优化在线决策单调性:二分队列
与上面的转移方程相反,若转移方程是 \(f+w\to f\),那么求 \(f_i\) 前就要先求出前面的所有 \(f\),无法采用上面的分治做法。
使用二分队列,队列维护若干三元组 \((l,r,p)\),表示 \([l,r]\) 的最优决策点是 \(p\),每次到 \(i\) 进行如下操作。
-
按照此时队首的 \(p\) 更新 \(f_i\),将队首的 \(l\) 修改为 \(i+1\),若 \(i=r\),弹出队首;
-
不断取出当前队尾的三元组,并进行如下操作:
-
若 \(r\) 处的决策 \(p\) 比 \(i\) 优,退出;
-
若 \(l\) 处的决策 \(i\) 比 \(p\) 优,则令 \(nowp=l\),弹出队尾;
-
剩余情况中,一定在 \([l,r]\) 中存在 \(x\) 使得 \([l,x]\) 的最优决策点是 \(p\),而 \([x+1,r]\) 的最优决策点是 \(i\),二分即可,修改队尾三元组为 \((l,x,p)\),退出;
-
-
将 \((nowp,n,i)\) 放入队尾。
这样时间复杂度 \(O(n\log n)\)。
例题
区间 DP 优化的例题是取 \(\min\) 的石子合并。
Luogu-P4360 CEOI 2004 锯木厂选址
预处理 \(s_i\) 表示将 \(i\) 作为第一个锯木厂,\([1,i]\) 的木头运送的花费。
状态是设 \(f_i\) 表示将 \(i\) 作为第二个锯木厂的花费,枚举第一个锯木厂 \(j\),结果分成三部分:\([1,j]\) 的花费 \(s_j\)、\([j+1,i]\) 的花费 \(w(j,i)\) 以及 \([i+1,n]\) 的花费,其中第三部分关于 \(i\) 是常量,且贡献函数 \(w(j,i)\) 满足四边形不等式,\(s\) 是异于 \(f\) 的函数,直接分治即可。
Luogu-P3515 POI 2011 Lightning Conductor
假设 \(p\) 可以取任意实数,最后再向上取整,设 \(f_i=a_i+p\),实际上要满足:
只考虑 \(j<i\) 或 \(j>i\),贡献函数满足四边形不等式,直接分治即可。
Luogu-P1912 NOI 2009 诗人小 G
转移方程是:
注意到 \(h(x)=|x-C|^P\) 是下凸函数,且 \(w(j,i)=(s_i+i)-(s_j+j)\) 满足四边形恒等式和区间包含单调性,那么贡献函数 \(h(w(j,i))\) 就满足四边形不等式,在线的转移使用二分队列。
题目中给出的 \(10^{18}\) 的限制不能通过单纯的与某个数取 \(\min\) 来满足,这样的操作有违决策单调性。
Luogu-P4767 IOI 2000 邮局
设 \(f_{k,i}\) 表示在 \(i\) 村庄设第 \(k\) 个邮局,\([1,i]\) 中每个村庄的最短距离之和,预处理 \(w(l,r)\) 表示只在 \(l,r\) 两个村庄设邮局,\([l,r]\) 中每个村庄的最短距离之和。
增加第 \(n+1\) 个村庄,坐标设为极大值,这样 \(f_{m+1,n+1}\) 就是答案(在 \(n+1\) 处设邮局一定只有自己是最近的)。
转移是:
考虑实际意义发现贡献函数满足四边形不等式,那么直接相邻两层分治转移即可,时间复杂度 \(O(nm\log n)\)。
CodeForces-868F Yet Another Minimization Problem *2500
贡献函数不好计算,使用上面的指针移动技巧即可,时间复杂度 \(O(nk\log n)\)。