题意:中文题。。不多说。
这题就是裸的Splay模板。。。但是出了一个问题。。导致我WA了12次(最后随机数对拍才找到)。。。
没插入一个点,把这个点旋转到根,然后找到根的左子树最靠右边的的,根的右子树最靠左边的点,然后相减处理下就好。。。
这里我是WA在,每把一个点旋转到根,忘记把表示根节点的那个点修改成根了。。。然后还RE了几发,原因是我写的是多组数据(测试是单组数据),然后我需要初始化,删除整个树的时候指针可能越界访问了。。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN=500010;
struct Node
{
Node *ch[2];
Node *fa;
int val;
}p[MAXN];
Node *null=&p[0];
int tot;
Node *root=null;
void Rotate(Node *x,int c)
{
Node *y=x->fa;
y->ch[!c]=x->ch[c];
if(x->ch[c]!=null)
x->ch[c]->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;
}
y->fa=x;
x->ch[c]=y;
if(y==root)
x=root;
}
void splay(Node *x,Node *f)
{
while(x->fa!=f)
{
if(x->fa->fa==f)
{
if(x->fa->ch[0]==x)
Rotate(x,1);
else
Rotate(x,0);
}
else
{
Node *y=x->fa,*z=y->fa;
if(z->ch[0]==y)
{
if(y->ch[0]==x)
Rotate(y,1),Rotate(x,1);
else
Rotate(x,0),Rotate(x,1);
}
else
{
if(y->ch[1]==x)
Rotate(y,0),Rotate(x,0);
else
Rotate(x,1),Rotate(x,0);
}
}
}
}
Node *new_Node(int val)
{
Node *rt=&p[++tot];
rt->ch[0]=rt->ch[1]=rt->fa=null;
rt->val=val;
return rt;
}
Node *root1=null;
Node *insert(Node *rt,int val)
{
if(rt==null)
{
rt=new_Node(val);
root1=rt;
return rt;
}
if(rt->val==val)
return rt;
if(rt->val>val)
{
rt->ch[0]=insert(rt->ch[0],val);
rt->ch[0]->fa=rt;
}
else
{
rt->ch[1]=insert(rt->ch[1],val);
rt->ch[1]->fa=rt;
}
return rt;
}
Node *Delete(Node *&rt)
{
if(rt==null)
return null;
if(rt->ch[0]!=null)
rt->ch[0]=Delete(rt->ch[0]);
if(rt->ch[1]!=null)
rt->ch[1]=Delete(rt->ch[1]);
rt=null;
return rt;
}
int main()
{
int n,i;
null->val=0;
//freopen("out.txt","r",stdin);
//freopen("out1.txt","w",stdout);
while(scanf("%d",&n)==1)
{
tot=0;
int x;
int ans=0;
for(i=1;i<=n;i++)
{
x=0;
scanf("%d",&x);
root1=null;
root=insert(root,x);
if(root1==null)
continue;
splay(root1,null);
root=root1; //根节点要随着相应的改变
if(i==1)
{
ans+=x;
continue;
}
Node *L,*R;
L=root1->ch[0];
while(L!=null&&L->ch[1]!=null)
L=L->ch[1];
R=root1->ch[1];
while(R!=null&&R->ch[0]!=null)
R=R->ch[0];
int temp1=1<<30,temp2=1<<30;
if(L!=null)
temp1=x-(L->val);
if(R!=null)
temp2=(R->val)-x;
int temp=min(temp1,temp2);
ans+=temp;
}
printf("%d\n",ans);
root=Delete(root1);
}
return 0;
}