AC日记——宠物收养所 bzoj 1208

1208

思路：

　　一棵splay树；

　　如果来者是宠物且树空，就将其加入树中；

　　如果树不空，则查找前驱后继，取最优，然后删点；

　　对人亦然；

　　注意边界和取模，最后的ans用long long其余用int即可；

来，上代码：

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;

#define maxn 80005
#define mod 1000000
#define INF 0x7fffffff

struct TreeNodeType {
    int w,key,opi,size,ch[];

    void destroy()
    {
        w=key=opi=size=ch[]=ch[]=;
    }

    void create(int x)
    {
        destroy();
        key=x;
    }
};
struct TreeNodeType tree[maxn<<];

int n,tot,flag,root;

long long ans;

inline int getson(int now)
{
    return tree[tree[now].opi].ch[]==now;
}

inline void updata(int now)
{
    tree[now].size=tree[now].w;
    if(tree[now].ch[]) tree[now].size+=tree[tree[now].ch[]].size;
    if(tree[now].ch[]) tree[now].size+=tree[tree[now].ch[]].size;
}

inline void rotate(int now)
{
    int opi=tree[now].opi,fopi=tree[opi].opi,pos=getson(now);
    tree[opi].ch[pos]=tree[now].ch[pos^];
    tree[tree[opi].ch[pos]].opi=opi,tree[now].opi=fopi;
    if(fopi) tree[fopi].ch[getson(opi)]=now;
    tree[opi].opi=now,tree[now].ch[pos^]=opi;
    updata(opi),updata(now);
}

void splay(int now)
{
    for(int opi;opi=tree[now].opi;rotate(now))
    {
        if(tree[opi].opi) rotate(getson(now)==getson(opi)?opi:now);
    }
    root=now;
}

void insert(int x)
{
    if(!root) tree[++tot].create(x),root=tot;
    else
    {
        int now=root,opi=;
        while()
        {
            if(tree[now].key==x)
            {
                tree[now].w++;
                tree[now].size++;
                splay(now);
                return ;
            }
            opi=now,now=tree[now].ch[x>tree[now].key];
            if(!now)
            {
                tree[++tot].create(x);
                tree[tot].opi=opi;
                tree[opi].ch[x>tree[opi].key]=tot;
                splay(tot);
                return ;
            }
        }
    }
}

inline int pre()
{
    if(tree[root].w>) return root;
    if(!tree[root].ch[]) return ;
    int now=tree[root].ch[];
    while(tree[now].ch[]) now=tree[now].ch[];
    return now;
}

inline int suc()
{
    if(tree[root].w>) return root;
    if(!tree[root].ch[]) return ;
    int now=tree[root].ch[];
    while(tree[now].ch[]) now=tree[now].ch[];
    return now;
}

void del()
{
    if(tree[root].w>)
    {
        tree[root].w--;
        tree[root].size--;
        return ;
    }
    if(!tree[root].ch[]&&!tree[root].ch[])
    {
        tree[root].destroy();
        root=;tree[root].destroy();return ;
    }
    if(tree[root].ch[]&&!tree[root].ch[])
    {
        int tmp=root;
        root=tree[root].ch[];
        tree[tmp].destroy();
        tree[root].opi=;
        return ;
    }
    if(!tree[root].ch[]&&tree[root].ch[])
    {
        int tmp=root;
        root=tree[root].ch[];
        tree[tmp].destroy();
        tree[root].opi=;
        return ;
    }
    int tmp=pre(),pos=root;
    tree[tmp].ch[]=tree[root].ch[];
    tree[tree[tmp].ch[]].opi=tmp;
    root=tree[root].ch[],tree[root].opi=;
    tree[pos].destroy();
    splay(tree[tmp].ch[]);
}

int main()
{
    scanf("%d",&n);int ty,x;
    while(n--)
    {
        scanf("%d%d",&ty,&x);
        if(!root)
        {
            flag=ty;
            insert(x);
        }
        else
        {
            if(ty==flag) insert(x);
            else
            {
                insert(x);
                int pr=pre(),su=suc(),to;
                if(!pr) ans+=abs(tree[su].key-x),to=su;
                else if(!su) ans+=abs(tree[pr].key-x),to=pr;
                else
                {
                    if(abs(tree[pr].key-x)<=abs(tree[su].key-x)) ans+=abs(tree[pr].key-x),to=pr;
                    else ans+=abs(tree[su].key-x),to=su;
                }
                del(),splay(to),del();
            }
        }
    }
    cout<<ans%mod;
    return ;
}