题意:求每个点之前离它差最小的数(两个相减的绝对值)
思路:Splay模板
如果父结点是根结点,那么只需要作一次左旋或者右旋即可。
如果父结点P,P的父结点G,如果二者转向相同,那么连续两次左旋或者右旋即可。
如果二者转向不同,那么向左旋后右旋或者先右旋后左旋。
#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
const int N=1e5+5;
int fa[N],key[N],ch[N][2],root,sz;//分别表示父亲结点,键值,左右孩子(0为左孩子,1为右孩子),根节点,节点数量
//新建一个结点
void Newnode(int &now,int father,int k){
now=++sz;
fa[now]=father;
key[now]=k;
ch[now][0]=ch[now][1]=0;
}
//旋转,kind为1为右旋,kind为0为左旋
void Rotate(int x,int kind){
int y=fa[x];
ch[y][!kind]=ch[x][kind];
fa[ch[x][kind]]=y;
//如果父节点不是根结点,则要和父节点的父节点连接起来
if(fa[y])
ch[ fa[y] ][ch[fa[y]][1]==y]=x;
fa[x]=fa[y];
ch[x][kind]=y;
fa[y]=x;
}
//Splay调整,将根为now的子树调整为goal
void Splay(int now,int goal){
while(fa[now]!=goal){
if(fa[ fa[now] ]==goal)
Rotate(now,ch[ fa[now] ][0]==now);
else{
int pre=fa[now],kind=ch[ fa[pre] ][0]==pre; //左儿子为1,右儿子为0
if(ch[pre][kind]==now){ //两个方向不同
Rotate(now,!kind);
Rotate(now,kind);
}
else{ //两个方向相同
Rotate(pre,kind);
Rotate(now,kind);
}
}
}
if(goal==0) root=now;
}
int Insert(int k){
int now=root;
while(ch[now][key[now]<k]){
//不重复插入
if(key[now]==k){
Splay(now,0);
return 0;
}
now=ch[now][key[now]<k];
}
if(key[now]==k){
Splay(now,0);
return 0;
}
Newnode(ch[now][k>key[now]],now,k);
//将新插入的结点更新至根结点
Splay(ch[now][k>key[now]],0);
return 1;
}
//找前驱,即左子树的最右结点
int get_pre(int x){
int tmp=ch[x][0];
if(tmp==0) return INF;
while(ch[tmp][1])
tmp=ch[tmp][1];
return key[x]-key[tmp];
}
//找后继,即右子树的最左结点
int get_next(int x){
int tmp=ch[x][1];
if(tmp==0) return INF;
while(ch[tmp][0])
tmp=ch[tmp][0];
return key[tmp]-key[x];
}
int main(){
int num,n;
while(scanf("%d",&n)!=EOF){
root=sz=0;
int ans=0;
for(int i=1;i<=n;i++){
scanf("%d",&num);
if(i==1){
ans+=num;
Newnode(root,0,num);
continue;
}
if(!Insert(num)) continue;
int minv1=get_next(root),minv2=get_pre(root);
ans+=min(minv1,minv2);
}
printf("%d\n",ans);
}
return 0;
}