Bzoj 1588: [HNOI2002]营业额统计(Splay 单点更新)

传送门:Bzoj 1588: [HNOI2002]营业额统计

---

题意:求每个点之前离它差最小的数(两个相减的绝对值)

---

思路: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;
}