bzoj1208 [HNOI2004] 宠物收养所 splay

Description

---

最近，阿Q开了一间宠物收养所。收养所提供两种服务：收养被主人遗弃的宠物和让新的主人领养这些宠物。每个领养者都希望领养到自己满意的宠物，阿Q根据领养者的要求通过他自己发明的一个特殊的公式，得出该领养者希望领养的宠物的特点值a（a是一个正整数，a<2^31），而他也给每个处在收养所的宠物一个特点值。这样他就能够很方便的处理整个领养宠物的过程了，宠物收养所总是会有两种情况发生：被遗弃的宠物过多或者是想要收养宠物的人太多，而宠物太少。 1． 被遗弃的宠物过多时，假若到来一个领养者，这个领养者希望领养的宠物的特点值为a，那么它将会领养一只目前未被领养的宠物中特点值最接近a的一只宠物。（任何两只宠物的特点值都不可能是相同的，任何两个领养者的希望领养宠物的特点值也不可能是一样的）如果有两只满足要求的宠物，即存在两只宠物他们的特点值分别为a-b和a+b，那么领养者将会领养特点值为a-b的那只宠物。 2． 收养宠物的人过多，假若到来一只被收养的宠物，那么哪个领养者能够领养它呢？能够领养它的领养者，是那个希望被领养宠物的特点值最接近该宠物特点值的领养者，如果该宠物的特点值为a，存在两个领养者他们希望领养宠物的特点值分别为a-b和a+b，那么特点值为a-b的那个领养者将成功领养该宠物。 一个领养者领养了一个特点值为a的宠物，而它本身希望领养的宠物的特点值为b，那么这个领养者的不满意程度为abs(a-b)。【任务描述】你得到了一年当中，领养者和被收养宠物到来收养所的情况，希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。

Solution

---

化简题意可以发现就是在一堆数字中间找最接近的数(绕)
那么开一棵splay，每次查找前驱后继删掉即可

删除x节点的操作可以看成是把x旋到根，再把x的后继旋到根的右子节点，删掉x让后继成为根即可
特判一下x是最大值的情况就直接删掉它

Code

---

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))
const int INF=0x3f3f3f3f;
const int MOD=1000000;
const int N=100005;
struct treeNode{int fa,key,son[2];}t[N<<2];
int cnt=0,root=0;
int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}
void update(int &x,int fa,int key) {t[x=++cnt]=(treeNode){fa,key,{0,0}};}
void rrotate(int x) {
    int y=t[x].fa; int z=t[y].fa;
    t[y].son[0]=t[x].son[1];
    if (t[x].son[1]) t[t[x].son[1]].fa=y;
    t[x].fa=z;
    t[x].son[1]=y; t[y].fa=x;
    if (z) {
        if (t[z].son[0]==y) t[z].son[0]=x;
        else t[z].son[1]=x;
    }
}
void lrotate(int x) {
    int y=t[x].fa; int z=t[y].fa;
    t[y].son[1]=t[x].son[0];
    if (t[x].son[0]) t[t[x].son[0]].fa=y;
    t[x].fa=z;
    t[x].son[0]=y; t[y].fa=x;
    if (z) {
        if (t[z].son[0]==y) t[z].son[0]=x;
        else t[z].son[1]=x;
    }
}
void splay(int x,int goal) {
    while (t[x].fa!=goal) {
        int y=t[x].fa; int z=t[y].fa;
        if (z==goal) {
            if (t[y].son[0]==x) rrotate(x);
            else lrotate(x);
        } else {
            if (t[z].son[0]==y&&t[y].son[0]==x) {
                rrotate(y);
                rrotate(x);
            } else if (t[z].son[0]==y&&t[y].son[1]==x) {
                lrotate(x);
                rrotate(x);
            } else if (t[z].son[1]==y&&t[y].son[0]==x) {
                rrotate(x);
                lrotate(x);
            } else if (t[z].son[1]==y&&t[y].son[1]==x) {
                lrotate(y);
                lrotate(x);
            }
        }
    }
    if (!goal) root=x;
}
void insert(int key) {
    if (!root) {
        update(root,0,key);
        return ;
    }
    int x=root;
    while (x) {
        if (t[x].key>key) {
            if (!t[x].son[0]) {
                update(t[x].son[0],x,key);
                break ;
            } else x=t[x].son[0];
        } else {
            if (!t[x].son[1]) {
                update(t[x].son[1],x,key);
                break ;
            } else x=t[x].son[1];
        }
    }
    splay(cnt,0);
}
void del(int x) {
    splay(x,0);
    int tmp=t[x].son[1];
    if (!tmp) {
        root=t[x].son[0];
        t[root].fa=0;
        return ;
    }
    while (t[tmp].son[0]) tmp=t[tmp].son[0];
    splay(tmp,root);
    t[tmp].son[0]=t[root].son[0];
    t[tmp].fa=0;
    t[t[root].son[0]].fa=tmp;
    root=tmp;
}
int get_pre(int key) {
    int x=root;
    int y=0;
    while (x) {
        if (t[x].key<=key) {
            y=x;
            x=t[x].son[1];
        } else x=t[x].son[0];
    }
    if (y) splay(y,0);
    return y;
}
int get_nex(int key) {
    int x=root;
    int y=0;
    while (x) {
        if (t[x].key>=key) {
            y=x;
            x=t[x].son[0];
        } else x=t[x].son[1];
    }
    if (y) splay(y,0);
    return y;
}
int main(void) {
    t[0].key=INF;
    int n=read();
    int now=2;
    int size=0;
    int ans=0;
    while (n--) {
        int opt=read();
        int x=read();
        if (!size) {
            now=opt;
            insert(x);
            size++;
        } else if (opt==now) {
            insert(x);
            size++;
        } else if (opt!=now) {
            size--;
            int pre=get_pre(x);
            int nex=get_nex(x);
            int PRE=std:: abs(x-t[pre].key);
            int NEX=std:: abs(x-t[nex].key);
            if (PRE<=NEX) {
                del(pre);
                ans=(ans+PRE)%MOD;
            } else {
                del(nex);
                ans=(ans+NEX)%MOD;
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}