BZOJ1588营业额统计 splay

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/liufengwei1/article/details/86647455

就一个找前驱后继。。。感觉map做也行

#include<bits/stdc++.h>
#define maxl 50010

using namespace std;

const int inf=0x3f3f3f3f;

int n,t1,t2,rt,size,ans;
int fa[maxl],num[maxl],a[maxl];
int tr[maxl][2];
namespace Splay
{
	void rotate(int x,int &k)
	{
		int y=fa[x],z=fa[y],l,r;
		if(tr[y][0]==x) l=0;else l=1;r=l^1;
		if(y==k) k=x;
		else if(tr[z][0]==y)
			tr[z][0]=x;
		else
			tr[z][1]=x;
		fa[x]=z;fa[y]=x;fa[tr[x][r]]=y;
		tr[y][l]=tr[x][r];tr[x][r]=y;
	}
	void splay(int x,int &k)
	{
		int y,z;
		while(x!=k)
		{
			y=fa[x];z=fa[y];
			if(y!=k)
			{
				if((tr[y][0]==x)^(tr[z][0]==y))
					rotate(x,k);
				else
					rotate(y,k);
			}
			rotate(x,k);
		}
	}
	void ins(int &k,int x,int last)
	{
		if(k==0)
		{
			size++;k=size;num[k]=x;fa[k]=last;
			splay(k,rt);
			return;
		}
		if(x<num[k]) ins(tr[k][0],x,k);
		else ins(tr[k][1],x,k);
	}
	void ask_before(int k,int x)
	{
		if(k==0) return;
		if(num[k]<=x){t1=num[k];ask_before(tr[k][1],x);}
		else ask_before(tr[k][0],x);
	}
	void ask_after(int k,int x)
	{
		if(k==0)
			return;
		if(num[k]>=x){t2=num[k];ask_after(tr[k][0],x);}
		else ask_after(tr[k][1],x);
	}
}
using namespace Splay;

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		t1=-inf;t2=inf;
		ask_before(rt,a[i]);
		ask_after(rt,a[i]);
		if(i!=1)
			ans+=min(a[i]-t1,t2-a[i]);
		else
			ans+=a[i];
		ins(rt,a[i],0);
	}
	printf("%d",ans);
	return 0;
}