bzoj #1208 宠物收养所（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)。【任务描述】你得到了一年当中，领养者和被收养宠物到来收养所的情况，希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。

Input

第一行为一个正整数n，n<=80000，表示一年当中来到收养所的宠物和领养者的总数。接下来的n行，按到来时间的先后顺序描述了一年当中来到收养所的宠物和领养者的情况。每行有两个正整数a, b，其中a=0表示宠物，a=1表示领养者，b表示宠物的特点值或是领养者希望领养宠物的特点值。（同一时间呆在收养所中的，要么全是宠物，要么全是领养者，这些宠物和领养者的个数不会超过10000个）

Output

仅有一个正整数，表示一年当中所有收养了宠物的领养者的不满意程度的总和mod 1000000以后的结果。

---

与bzoj1588类似，也是用splay去做。
不同之处是，需要记录当前树是宠物树还是顾客树，有一点细节问题。
还需要维护多一个del操作，具体是把要删除的点splay到根上，若其左右儿子节点都有，则将左子树中最大的点splay到根上，将要删除的店的右儿子节点挂到该节点的右儿子上。

#include<cstdio>
#include<algorithm>
#include<cstring>
#define maxn 80050
#define mod 1000000
#define INF 2000000010
using namespace std;
struct node
{
    int f,s[2],w;
}a[maxn*2];
int tk,ts,dfn,p,q,pnum,qnum,hz,hnum,root,ans=0,tmp,n,k,x;//treekind,k=0 pet,k=1 customer;treesize
inline void rotate(int x,int k)
{
    int y=a[x].f,z=a[y].f;
    a[y].s[!k]=a[x].s[k];
    if (a[x].s[k]!=-1) a[a[x].s[k]].f=y;
    a[x].f=z;
    if (z!=-1) a[z].s[y==a[z].s[1]]=x;
    a[x].s[k]=y,a[y].f=x;
}
inline void splay(int x)
{
    while (a[x].f!=-1)
    {
        int y=a[x].f,z=a[y].f;
        if (z==-1) {rotate(x,x==a[y].s[0]);continue;}
        if (y==a[z].s[0])
        {
            if (x==a[y].s[0]) rotate(y,1),rotate(x,1);
            else rotate(x,0),rotate(x,1);
        }
        else
        {
            if (x==a[y].s[1]) rotate(y,0),rotate(x,0);
            else rotate(x,1),rotate(x,0);
        }
    }
    root=x;
}
inline void del(int x)
{
    splay(x);
    if (a[x].s[0]==-1) {root=a[x].s[1],a[a[x].s[1]].f=-1;return;}
    if (a[x].s[1]==-1) {root=a[x].s[0],a[a[x].s[0]].f=-1;return;}
    int y=a[x].s[0];
    while (a[y].s[1]!=-1) y=a[y].s[1];
    a[a[x].s[0]].f=-1,splay(y),a[y].s[1]=a[x].s[1],a[a[x].s[1]].f=y,root=y;
}
inline void insert(int x,int now)
{
    if (now<a[x].w) 
    {
        if (a[x].s[0]!=-1) insert(a[x].s[0],now);
        else a[x].s[0]=++dfn,a[dfn].f=x,a[dfn].s[0]=a[dfn].s[1]=-1,a[dfn].w=now;
    }
    else
    {
        if (a[x].s[1]!=-1) insert(a[x].s[1],now);
        else a[x].s[1]=++dfn,a[dfn].f=x,a[dfn].s[0]=a[dfn].s[1]=-1,a[dfn].w=now;
    }
}
inline void pre(int x)
{
    tmp=root,p=-1,pnum=INF;
    while (1)
    {
        if (tmp==-1) return;
        if (a[tmp].w<x&&(x-a[tmp].w<pnum)) p=tmp,pnum=x-a[tmp].w;
        if (a[tmp].w<x) tmp=a[tmp].s[1];else tmp=a[tmp].s[0];
    }
}
inline void suc(int x)
{
    tmp=root,q=-1,qnum=INF;
    while (1)
    {
        if (tmp==-1) return;
        if (a[tmp].w>x&&(a[tmp].w-x<pnum)) q=tmp,qnum=a[tmp].w-x;
        if (a[tmp].w<x) tmp=a[tmp].s[1];else tmp=a[tmp].s[0];
    }
}
int main()
{
    scanf("%d",&n),ts=0;
    for (int i=1;i<=n;i++)
    {
        scanf("%d%d",&k,&x);
        if (ts==0) 
        {
            root=ts=1,a[1].f=a[1].s[0]=a[1].s[1]=-1,a[1].w=x,tk=k,dfn=1;
            continue;
        }
        if (tk==k) {insert(root,x),ts++;continue;}
        pre(x),suc(x),hz=-1,hnum=INF+1;
        if (p!=-1&&pnum<hnum) hz=p,hnum=pnum;
        if (q!=-1&&qnum<hnum) hz=q,hnum=qnum;
        ans=(ans+hnum)%mod,del(hz),ts--;
    }
    printf("%d",ans);
}