【HNOI2002】营业额统计【重做-Splay做法】

之前用sbt写了这个题现在自己开发了一套Splay的模板继续拿这道题试试手

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXINT 0x7fffffff
using namespace std;
struct splay
{
	splay *ch[2],*fa;//ch[0]为左子树ch[1]为右子树fa为父节点
	int data,cnt,size;//data节点数值cnt重复元素个数size树的大小
}*root,*no;
int ans;
int n,d;
void calc(splay *x)
{
	x->size=0;
	if (x->ch[0]) x->size+=x->ch[0]->size;
	if (x->ch[1]) x->size+=x->ch[1]->size;
	x->size+=1;
}
void rot(splay *x,bool flag)//flag=0左旋flag=1右旋 节省编程复杂度的小优化
{
	splay *y=x->fa;
	y->ch[!flag]=x->ch[flag];
    if (x->ch[flag]!=NULL) x->ch[flag]->fa=y;
	x->fa=y->fa;
	if (y->fa!=NULL)
	    if (y->fa->ch[0]==y) y->fa->ch[0]=x;
	    else y->fa->ch[1]=x;
	x->ch[flag]=y;
	y->fa=x;
	if (y==root) root=x;
}
void Splay(splay *x,splay *f)//将节点X提到节点f下 (地位与sbt Maintain()相同的操作0-0)
{
	while (x->fa!=f)
	{
		if (x->fa->fa==f)
		{
			if (x->fa->ch[0]==x) rot(x,1);
		    else rot(x,0);
		}
		else
		{
			splay *y=x->fa,*z=y->fa;
			if (z->ch[0]==y)
			    if (y->ch[0]==x)
			        rot(y,1),rot(x,1);
			    else
			        rot(x,0),rot(x,1);
			else
			    if (y->ch[0]==x)
			        rot(x,1),rot(x,0);
			    else
			        rot(y,0),rot(x,0);
		}
	}
	calc(x);
	calc(f);
}
void insert(int data)
{
	splay *n=root,*x=NULL;
	while (n!=NULL)
	{
		x=n;
		if (data>=n->data) n=n->ch[1];
		else n=n->ch[0];
	}
	n=new splay;
	n->ch[0]=n->ch[1]=NULL;
	n->size=1;
	n->data=data;
	n->cnt=1;
	n->fa=x;
	if (x==NULL) root=n;
	else
	if (data>=x->data) x->ch[1]=n;
	else x->ch[0]=n;
	if (n!=root) Splay(n,root);
}
splay* find(int data)
{
	if (root==NULL) return NULL;
	splay *x=root,*y=NULL;
	while (x!=NULL)
	{
		if (data>x->data)
		    y=x,x=x->ch[1];
		else
		if (data<x->data)
		    y=x,x=x->ch[0];
		else
		    return y;
	}
	return NULL;
}
splay* getmax()
{
	if (root==NULL) return NULL;
	splay *x=root;
	while (x->ch[1]) x=x->ch[1];
	return x;
}
splay* getmin()
{
	if (root==NULL) return NULL;
	splay *x=root;
	while (x->ch[0]) x=x->ch[0];
	return x;
}
int rank(int data)
{
	splay *x=find(data);
	Splay(x,root);
	if (root->ch[0]==x) return x->ch[0]->size+1;
	else
	return root->ch[0]->size+1+x->ch[0]->size+1;
}
splay* select(int k)
{
	splay *x=root;
	while (x)
	{
		if (k>x->ch[0]->size+1)
		{
			k-=x->ch[0]->size+1;
			x=x->ch[1];
		}
		else
		if (k<x->ch[0]->size+1)
		{
			x=x->ch[0];
		}
		else
		    return x;
	}
}
splay* pred(splay *x,splay *y,int data)
{
	if (x==NULL) return y;
	if (data==x->data) return y;
	if (data>x->data) return pred(x->ch[1],x,data);
	else return pred(x->ch[0],y,data);
}
splay* succ(splay *x,splay *y,int data)
{
	if (x==0) return y;
	if (x->data==data) return x;
	if (x->data>data) return succ(x->ch[0],x,data);
	else return succ(x->ch[1],y,data);
}
void remove(int data)
{
	splay *x=find(data);
	Splay(x,root);
	splay *t=succ(root,no,data);
	if (t!=no)
	{
		Splay(t,x);
		if (root->ch[1]==x)
		{
			root->ch[1]=t;
			t->ch[0]=x->ch[0];
		}
		else
		{
			root->ch[0]=t;
			t->ch[0]=x->ch[0];
		}
	}
	else
	{
		if (root->ch[1]==x)
		{
			root->ch[1]=t;
			t->ch[0]=x->ch[0];
		}
		else
		{
			root->ch[0]=t;
			t->ch[0]=x->ch[0];
		}
	}
}
int main()
{
	freopen("turnover.in","r",stdin);
	freopen("turnover.out","w",stdout);
	scanf("%d",&n);
	no=NULL;
	for (int i=1;i<=n;i++)
	{
		if (scanf("%d",&d)==EOF) d=0;
		splay *a=succ(root,no,d);
		splay *b=pred(root,no,d);
		int tmpa;
		int tmpb;
		if (a==no) tmpa=MAXINT;
		else tmpa=abs(d-a->data);
		if (b==no) tmpb=MAXINT;
		else tmpb=abs(d-b->data);
		if (a||b) ans+=min(tmpa,tmpb);
		else ans+=d;
		insert(d);
    }
    cout<<ans;
}