题意
平面上有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;
}