• 斜率优化DP总结


    前言

    (本文中的图片都由(WPS)出品)
    (DP)(OI) 中重要的一部分
    一般来说,因为 (DP) 会把之前的结果保存下来,所以时间复杂度还是比较优秀的
    但是在某些情况下,时间复杂度仍然超出了题目的限制,这是我们就要考虑对其进行优化
    (DP) 的优化一般从状态、决策、转移三个方面去考虑
    而斜率优化则是对决策进行优化的一种方法
    它适用于类似 (f[i]=min/max(a[i] imes b[j]+c[i]+d[j])) 的方程

    例题(洛谷 P4072 [SDOI2016]征途 )

    题目描述

    (Pine) 开始了从 (S) 地到 (T) 地的征途。

    (S)地到(T)地的路可以划分成 (n) 段,相邻两段路的分界点设有休息站。

    (Pine)计划用(m)天到达(T)地。除第(m)天外,每一天晚上(Pine)都必须在休息站过夜。所以,一段路必须在同一天中走完。

    (Pine)希望每一天走的路长度尽可能相近,所以他希望每一天走的路的长度的方差尽可能小。

    帮助(Pine)求出最小方差是多少。

    设方差是(v),可以证明,(v imes m^2)是一个整数。为了避免精度误差,输出结果时输出(v imes m^2)

    输入格式

    第一行两个数 (n)(m)

    第二行 (n) 个数,表示 (n) 段路的长度

    输出格式

    一个数,最小方差乘以 (m^2) 后的值

    输入输出样例

    输入 #1

    5 2
    1 2 5 8 6

    输出 #1

    36

    说明/提示

    对于 (30\%) 的数据,(1 le n le 10)

    对于 (60\%) 的数据,(1 le n le 100)

    对于 (100\%) 的数据,(1 le n le 3000)

    保证从 (S)(T) 的总路程不超过 (30000)

    分析

    要对一个状态转移方程进行优化,首先要把最朴素的方程写出来

    在本题中,稍加推导即可写出时间复杂度为 (O(m imes n^2))的状态转移方程

    [s^2 imes m^2=frac{(v_1-overline{v})^2+(v_2-overline{v})^2+...+(v_m-overline{v})^2}m imes m^2 ]

    [s^2 imes m^2=((v_1-overline{v})^2+(v_2-overline{v})^2+...+(v_m-overline{v})^2) imes m ]

    [s^2 imes m^2=m imes sum_{i=1}^mv_i^2+m^2 imes overline {v}^2-2 imes m imes overline {v} imes sum[n] ]

    又因为 (overline{v}=frac{sum[n]}{m})

    所以

    [s^2 imes m^2=m imes sum_{i=1}^mv_i^2-sum[n] imes sum[n] ]

    后面的值是固定的,所以我们只需要让前面的值最小化即可

    我们设(f[i][j])为前(i)天分成(j)段所得到的最小值

    那么就有

    [f[i][k]=min(f[j][k-1]+(sum[i]-sum[j])^2) ]

    展开就有

    [f[i][k]=f[j][k-1]+sum[i]^2+sum[j]^2-2 imes sum[i] imes sum[j] ]

    接下来就是本文的重点:如何用斜率优化这类方程

    首先,你需要掌握一次函数 (y=kx+b) 的图像和性质

    这应该问题不大

    下面我们就要对方程进行移项,使其变成易于优化的形式

    [f[j][k-1]+sum[j]^2=2 imes sum[i] imes sum[j]+f[i][k]-sum[i]^2 ]

    我们发现,这和一次函数的解析式完全吻合

    我们把(f[j][k-1]+sum[j]^2)看成(y)

    (2 imes sum[i])看成(k)

    (sum[j])看成(x)

    (f[i][k]-sum[i]^2)看成(b)

    这样,对于每一个(i)来说,直线的(k)是确定的

    我们要使(f[i][k])最小,也就是要使(b)最小

    我们可以把所有的(j)想象成空间坐标为((sum[j],f[j][k-1]+sum[j]^2))中的点

    知道了斜率,知道了直线上的点,那么这条直线就确定了

    那么我们考虑什么样的点使直线的(b)最小

    直线(l)是我们要移动的直线,平面中的点是可以转移的(j)

    我们会发现当当前点和后一个点形成的直线的斜率恰好大于直线(l)的斜率时,由当前点转移决策是最优的

    在这里要特别强调一下:本题中斜率(k) 和 横坐标 (x) 均为单调递增的,对于 (k)(x)不单调递增的情况,处理方式不同

    这就是代码里面的

    while(head<tail && xl(q[head],q[head+1])<2*sum[j]) head++;
    

    我们再去考虑什么样的点肯定不会对结果产生贡献

    上面的图中(2)号节点是无论如何也不会更新其它节点的

    因为(1)号节点或(3)号节点总会比它更优

    这就是代码里的

    while(head<tail && xl(j,q[tail-1])>=xl(j,q[tail])) tail--;
    

    整个过程就相当于维护了一个下凸包

    在求斜率的函数中,我们要判掉 (x) 相等的情况,在某些时候,还要判掉 (y) 相等的情况

    double X(int id){
    	return (double)sum[id];
    }
    double Y(int id){
    	return (double)(g[id]+sum[id]*sum[id]);
    }
    double xl(int i,int j){
    	if(std::fabs(X(i)-X(j))<eps){
    		if(std::fabs(Y(i)-Y(j))<eps) return 0;
    		else if(Y(i)>Y(j)) return 1e18;
    		else return -1e18;
    	}
    	return (Y(i)-Y(j))/(X(i)-X(j));
    }
    

    拓展一:斜率不单调但x单调

    如果斜率不是单调递增,我们就不能从前面清空队列直接转移

    比如上面这幅图如果在遇到直线 (m) 时一直从前清空队列的话那么就会把(3)号决策点弹出队列

    但是如果之后遇到一个斜率比较小的直线(l)那么就不能转移到最优解

    典型的例题是 洛谷P5785 [SDOI2012]任务安排

    朴素的状态转移方程为

    $ f[i] = f[j] + (sumc[i] - sumc[j]) * sumt[i] + s * (sumc[n] - sumc[j]);$

    在这一道题中,作为斜率的 (sumt) 不再单调

    但是 (x) 之仍然是单调的

    所以我们可以用维护一个斜率单调的队列

    每次在队列中二分答案

    值得一提的是,出题人精心准备了卡精度的数据,所以我们要把二分时的除法改为乘法

    代码

    #include <cstdio>
    #define rg register
    inline int read() {
        rg int x = 0, fh = 1;
        rg char ch = getchar();
        while (ch < '0' || ch > '9') {
            if (ch == '-')
                fh = -1;
            ch = getchar();
        }
        while (ch >= '0' && ch <= '9') {
            x = (x << 1) + (x << 3) + (ch ^ 48);
            ch = getchar();
        }
        return x * fh;
    }
    typedef long long ll;
    const int maxn = 1e6 + 5;
    int t[maxn], c[maxn], n, s, q[maxn], head, tail;
    ll f[maxn], sumt[maxn], sumc[maxn];
    double Y(int i) { return (double)(f[i] - s * sumc[i]); }
    double X(int i) { return (double)sumc[i]; }
    double xl(int i, int j) {
        if (X(i) == X(j)) {
            if (Y(i) > Y(j))
                return 1e18;
            else
                return -1e18;
        }
        return (double)(Y(i) - Y(j)) / (X(i) - X(j));
    }
    int ef(double now) {
        int l = head, r = tail, mids;
        while (l < r) {
            mids = (l + r) >> 1;
            if ((X(q[mids]) > X(q[mids + 1]) &&
                 Y(q[mids]) - Y(q[mids + 1]) < now * (X(q[mids]) - X(q[mids + 1]))) ||
                (X(q[mids]) < X(q[mids + 1]) &&
                 (Y(q[mids + 1]) - Y(q[mids]) < now * (X(q[mids + 1]) - X(q[mids])))))
                l = mids + 1;
            else
                r = mids;
        }
        return q[l];
    }
    int main() {
        n = read(), s = read();
        for (rg int i = 1; i <= n; i++) {
            t[i] = read();
            c[i] = read();
            sumt[i] = sumt[i - 1] + t[i];
            sumc[i] = sumc[i - 1] + c[i];
        }
        head = tail = 1;
        for (rg int i = 1; i <= n; i++) {
            rg int wz = ef((double)(sumt[i]));
            f[i] = f[wz] + (sumc[i] - sumc[wz]) * sumt[i] + s * (sumc[n] - sumc[wz]);
            while (head < tail && xl(i, q[tail - 1]) >= xl(i, q[tail])) tail--;
            q[++tail] = i;
        }
        printf("%lld
    ", f[n]);
        return 0;
    }
    

    扩展二、x不单调

    没有找到 (x) 不单调但是 (k) 单调的题

    但是却有一道 (x) 不单调 (k) 也不单调的题洛谷P4655 [CEOI2017]Building Bridges

    对于这道题,我们同样可以写出最朴素的方程

    (f[i]f[j]+(h[i]-h[j])*(h[i]-h[j])+sum[i-1]-sum[j])

    神奇的一点是 (h) 数组既作为直线的斜率又作为 (x)

    而且 (h) 并不单调

    这是我们就不能再用单调队列去维护,因为凸包的形状在不断改变

    本题可以用李超线段树或者平衡树动态维护凸包解决

    但是还有一种 (CDQ) 分治离线处理的方法

    直接人为地排出单调性,像普通单调队列那样维护就可以了

    如果左侧斜率递增,并且左侧编号小于右侧,那么可以通过单调队列维护左侧的凸包来更新右侧答案

    并且这样一定能够遍历出每个节点的所有决策点

    代码

    #include<cstdio>
    #include<cstring>
    #include<algorithm>
    #include<cstring>
    #include<cmath>
    #define rg register
    inline int read(){
    	rg int x=0,fh=1;
    	rg char ch=getchar();
    	while(ch<'0' || ch>'9'){
    		if(ch=='-') fh=-1;
    		ch=getchar();
    	}
    	while(ch>='0' && ch<='9'){
    		x=(x<<1)+(x<<3)+(ch^48);
    		ch=getchar();
    	}
    	return x*fh;
    }
    const int maxn=1e5+5;
    const double eps=1e-6;
    typedef long long ll;
    int n,h[maxn],w[maxn];
    ll f[maxn],sum[maxn];
    double X(int id){
    	return (double)h[id];
    }
    double Y(int id){
    	return (double)(f[id]+1LL*h[id]*h[id]-sum[id]);
    }
    double xl(int i,int j){
    	if(std::fabs(X(i)-X(j))<eps){
    		if(std::fabs(Y(i)-Y(j))<eps) return 0;
    		else if(Y(i)>Y(j)) return 1e18;
    		else return -1e18;
    	} else {
    		return (Y(i)-Y(j))/(X(i)-X(j));
    	}
    }
    bool cmp(int aa,int bb){
    	return h[aa]<h[bb];
    }
    int tmp[maxn],p[maxn],q[maxn];
    void solve(int l,int r){
    	if(l==r) return;
    	rg int mids=(l+r)>>1,head=l-1,tail=mids;
    	for(rg int i=l;i<=r;i++){
    		if(p[i]<=mids) tmp[++head]=p[i];
    		else tmp[++tail]=p[i];
    	}
    	for(rg int i=l;i<=r;i++){
    		p[i]=tmp[i];
    	}
    	solve(l,mids);
    	head=1,tail=0;
    	for(rg int i=l;i<=mids;i++){
    		while(head<tail && xl(p[i],q[tail-1])>=xl(p[i],q[tail])) tail--;
    		q[++tail]=p[i];
    	}
    	for(rg int i=mids+1;i<=r;i++){
    		while(head<tail && xl(q[head+1],q[head])<=2.0*h[p[i]]) head++;
    		f[p[i]]=std::min(f[p[i]],f[q[head]]+1LL*(h[p[i]]-h[q[head]])*(h[p[i]]-h[q[head]])+sum[p[i]-1]-sum[q[head]]);
    	}
    	solve(mids+1,r);
    	head=l,tail=mids+1;
    	for(rg int i=l;i<=r;i++){
    		if(tail>r || (head<=mids && h[p[head]]<h[p[tail]])) tmp[i]=p[head++];
    		else tmp[i]=p[tail++];
    	}
    	for(rg int i=l;i<=r;i++){
    		p[i]=tmp[i];
    	}
    }
    int main(){
    	memset(f,0x3f,sizeof(f));
    	n=read();
    	for(rg int i=1;i<=n;i++){
    		h[i]=read();
    		p[i]=i;
    	}
    	for(rg int i=1;i<=n;i++){
    		w[i]=read();
    		sum[i]=sum[i-1]+w[i];
    	}
    	std::sort(p+1,p+1+n,cmp);
    	f[1]=0;
    	solve(1,n);
    	printf("%lld
    ",f[n]);
    	return 0;
    }
    
    

    总结

    虽然斜率优化看起来很难,但是熟能生巧,多打几遍就能掌握

  • 相关阅读:
    Git查询
    HDU-3038 How Many Answers Are Wrong 并查集
    CodeForcesEducationalRound40-D Fight Against Traffic 最短路
    HDU-6109 数据分割 并查集(维护根节点)
    ZOJ-3261 Connections in Galaxy War 并查集 离线操作
    AtCoderBeginner091-C 2D Plane 2N Points 模拟问题
    HDU-1878 欧拉回路 欧拉回路
    [笔记-图论]Floyd
    [笔记-图论]Bellman-Ford
    [笔记-图论]Dijkstra
  • 原文地址:https://www.cnblogs.com/liuchanglc/p/13860779.html
Copyright © 2020-2023  润新知