AtCoder Regular Contest 073 F - Many Moves 线段树优化dp

题意

平面上有n个点，一开始有两个人，分别在点a和点b，移动速度为1距离/单位时间。有q个操作，每次要求要求将其中一个人移动到某一个特定的位置。问最少的总花费时间是多少。
n,q<=200000

分析

设f[i]表示在做完第i次操作后，一个人在x[i]，另一个人在x[i-1]时的最小花费是多少，sum[i]表示一个人从x[1]开始，按顺序做完[1,i]操作的花费时间。
不难得到转移f[i]=min(f[j]+sum[i-1]-sum[j]+|x[j-1]-x[i]|)。
把绝对值拆开来讨论，然后用两棵线段树来维护最小值即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=200005;
const LL inf=(LL)1e16;

int n,m,a,b;
LL sum[N],x[N],f[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

struct Segment_Tree
{
    struct tree{LL mn;}t[N*4];

    void build(int d,int l,int r)
    {
        t[d].mn=inf;
        if (l==r) return;
        int mid=(l+r)/2;
        build(d*2,l,mid);build(d*2+1,mid+1,r);
    }

    void ins(int d,int l,int r,int x,LL y)
    {
        if (l==r) {t[d].mn=y;return;}
        int mid=(l+r)/2;
        if (x<=mid) ins(d*2,l,mid,x,y);
        else ins(d*2+1,mid+1,r,x,y);
        t[d].mn=min(t[d*2].mn,t[d*2+1].mn);
    }

    LL query(int d,int l,int r,int x,int y)
    {
        if (l==x&&r==y) return t[d].mn;
        int mid=(l+r)/2;
        if (y<=mid) return query(d*2,l,mid,x,y);
        else if (x>mid) return query(d*2+1,mid+1,r,x,y);
        else return min(query(d*2,l,mid,x,mid),query(d*2+1,mid+1,r,mid+1,y));
    }
}t1,t2;

void dp()
{
    t1.ins(1,1,m,x[0],f[1]-x[0]-sum[1]);
    t2.ins(1,1,m,x[0],f[1]+x[0]-sum[1]);
    for (int i=2;i<=n;i++)
    {
        f[i]=min(t1.query(1,1,m,1,x[i])+x[i],t2.query(1,1,m,x[i],m)-x[i])+sum[i-1];
        t1.ins(1,1,m,x[i-1],f[i]-x[i-1]-sum[i]);
        t2.ins(1,1,m,x[i-1],f[i]+x[i-1]-sum[i]);
    }
}

int main()
{
    m=read();n=read();a=read();b=read();
    for (int i=1;i<=n;i++) x[i]=read();
    for (int i=2;i<=n;i++) sum[i]=sum[i-1]+abs(x[i]-x[i-1]);
    LL ans=inf;
    f[1]=abs(a-x[1]);x[0]=b;
    t1.build(1,1,m);t2.build(1,1,m);
    dp();
    for (int i=1;i<=n;i++) ans=min(ans,f[i]+sum[n]-sum[i]);
    f[1]=abs(b-x[1]);x[0]=a;
    t1.build(1,1,m);t2.build(1,1,m);
    dp();
    for (int i=1;i<=n;i++) ans=min(ans,f[i]+sum[n]-sum[i]);
    printf("%lld",ans);
    return 0;
}