平衡树插入和查询。
【代码】
#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 50005
#define M 200005
#define INF 1e9+1
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;
int read()
{
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
return x*f;
}
int n,mn,mx,rt,sz,ans;
int son[N][2],fa[N],num[N];
void Ask_Min(int k,int x)
{
if(!k) return;
if(num[k]<=x)
{
mn=num[k];
Ask_Min(son[k][1],x);
}
else Ask_Min(son[k][0],x);
}
void Ask_Max(int k,int x)
{
if(!k) return;
if(num[k]>=x)
{
mx=num[k];
Ask_Max(son[k][0],x);
}
else Ask_Max(son[k][1],x);
}
void Rotate(int x,int &k)
{
int y=fa[x],z=fa[y],l,r;
l=(son[y][0]!=x);r=l^1;
if(y==k) k=x;
else son[z][son[z][0]!=y]=x;
fa[x]=z;fa[y]=x;fa[son[x][r]]=y;
son[y][l]=son[x][r];son[x][r]=y;
}
void Splay(int x,int &k)
{
while(x!=k)
{
int y=fa[x],z=fa[y];
if(y!=k)
{
if((son[z][0]==y)^(son[y][0]==x)) Rotate(x,k);
else Rotate(y,k);
}
Rotate(x,k);
}
}
void Insert(int &k,int x,int Last)
{
if(!k)
{
k=++sz;fa[k]=Last;
num[k]=x;Splay(k,rt);
return;
}
if(x<num[k]) Insert(son[k][0],x,k);
else Insert(son[k][1],x,k);
}
int main()
{
n=read();
for(int i=1;i<=n;i++)
{
static int x;
x=read();
mn=-INF,mx=INF;
Ask_Min(rt,x);
Ask_Max(rt,x);
ans+=(i==1)?x:min(x-mn,mx-x);
Insert(rt,x,0);
}
printf("%d\n",ans);
return 0;
}