BZOJ 1588：splay tree

#include"cstdio"
#include"queue"
#include"cmath"
#include"stack"
#include"iostream"
#include"algorithm"
#include"cstring"
#include"queue"
#include"map"
#include"vector"
#define ll long long
#define mems(a,b) memset(a,b,sizeof(a))

using namespace std;
const int MAXN = 1e5+50;
const int MAXE = 200500;
const int INF = 0x3f3f3f;

int pre[MAXN],val[MAXN],ch[MAXN][2];///父亲结点、值、孩子节点（0左1右）
int tot,root;///总结点数、根节点
///中序遍历为所维护数列

void newnode(int &pos,int fa,int w){
pos=++tot;
pre[pos]=fa;
val[pos]=w;
ch[pos][0]=ch[pos][1]=0;
}

void Rotate(int x,int kind){///0左1右
int fa=pre[x];
ch[fa][!kind]=ch[x][kind];
pre[ch[x][kind]]=fa;

if(pre[fa]) ch[pre[fa]][ch[pre[fa]][1]==fa]=x;
pre[x]=pre[fa];

ch[x][kind]=fa;
pre[fa]=x;
}

void Splay(int r,int goal){///将r结点旋转至goal下方
while(pre[r]!=goal){
if(pre[pre[r]]==goal) Rotate(r,ch[pre[r]][0]==r);
else{
int fa=pre[r];
int kind=ch[pre[fa]][0]==fa;
if(ch[fa][kind]==r){///左右交替
Rotate(r,!kind);
Rotate(r,kind);
}
else{               ///方向一致
Rotate(fa,kind);
Rotate(r,kind);
}
}
}
if(!goal) root=r;///goal为不存在结点时，r为变为根节点
}

bool Insert(int key){
int r=root;
while(ch[r][val[r]<key]){
if(val[r]==key){///相同值只插入一次
Splay(r,0);
return false;
}
r=ch[r][val[r]<key];
}
newnode(ch[r][val[r]<key],r,key);
Splay(ch[r][val[r]<key],0);
return true;
}
///BST查询复杂度log(n)
int get_pre(int x){///寻找比X大但最接近X的数
int t=ch[x][0];
if(!t) return INF;
while(ch[t][1]) t=ch[t][1];
return val[x]-val[t];
}

int get_next(int x){///寻找比X小但最接近X的数
int t=ch[x][1];
if(!t) return INF;
while(ch[t][0]) t=ch[t][0];
return val[t]-val[x];
}

int main(){
int n;
//freopen("in.txt","r",stdin);
while(~scanf("%d",&n)){
root=tot=0;
int ans=0;
for(int i=1;i<=n;i++){
int x;
if(scanf("%d",&x)==EOF) x=0;
if(i==1){
ans+=x;
newnode(root,0,x);
continue;
}
if(!Insert(x)) continue;

ans+=min(get_pre(root),get_next(root));
}
cout<<ans<<endl;
}
return 0;
}