【codevs1285】【BZOJ1208】宠物收养所，splay练习

传送门1
传送门2
写在前面：我自己也会被写进机房大事记……
思路：很裸的splay练习，只涉及查前驱后继和删除操作，加一个标记记录现在是宠物多还是人多即可
注意：
1.查询和调用前驱后继时可能两者同时存在，记得判断一下
2.不知道为什么用bool型做树的标记时总是自己莫名改值，害我调了好久
代码：

#include<bits/stdc++.h>
#define mod 1000000
int n,x,tot,root,ans;
int hand,flag;
struct os
{
    int data,fa,left,right,sz;
}a[100010];
int in()
{
    int f=1,t=0;
    char ch=getchar();
    while (ch>'9'||ch<'0')
    {
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9') t=t*10+ch-'0',ch=getchar();
    return f*t;
}
void made(int x)
{
    a[++tot].data=x;
    a[tot].sz=1;
}
void ct(int now)
{
    a[now].sz=a[now].left+a[now].right+1;
}
void rorate(int now,bool flag)
{
    int pa=a[now].fa;
    if (flag)
    {
        a[pa].left=a[now].right;
        if (a[now].right)a[a[now].right].fa=pa;
        if (a[pa].fa)
        {
            if (a[a[pa].fa].left==pa) a[a[pa].fa].left=now;
            else a[a[pa].fa].right=now;
        }
        a[now].fa=a[pa].fa;
        a[pa].fa=now;
        a[now].right=pa;
    }
    else
    {
        a[pa].right=a[now].left;
        if (a[now].left)a[a[now].left].fa=pa;
        if (a[pa].fa) 
        { 
            if (a[a[pa].fa].left==pa) a[a[pa].fa].left=now;
            else a[a[pa].fa].right=now;
        } 
        a[now].fa=a[pa].fa;
        a[pa].fa=now;
        a[now].left=pa;
    }
    ct(pa);
    ct(now);
}
void splay(int x,int goal)
{
    int y;
    while (a[x].fa!=goal)
    {
        y=a[x].fa;
        if (a[y].fa==goal)
        {
            if (a[y].left==x) rorate(x,1);
            else rorate(x,0);
        }
        else if (y==a[a[y].fa].left)
        {
            if (a[y].left==x) rorate(y,1);
            else rorate(x,0);
            rorate(x,1);
        }
        else
        {
            if (a[y].right==x) rorate(y,0);
            else rorate(x,1);
            rorate(x,0);
        }
    }
    if (!goal) root=x;
}
void insert(int x)
{
    made(x);a[tot].data=x;
    if (!root) {root=tot;return;}
    int now=root;
    while (now)
    {
        if (x<a[now].data)
        {
            if (!a[now].left) {a[now].left=tot;a[tot].fa=now;break;}
            else now=a[now].left;
        }
        else
        {
            if (!a[now].right) {a[now].right=tot;a[tot].fa=now;break;}
            else now=a[now].right;
        }
    }
    splay(tot,0);
}
int find(int x)
{
    int now=root;
    while (now)
    {
        if (a[now].data==x) return now;
        else if (a[now].data>x) now=a[now].left;
        else now=a[now].right;
    }
    return 0;
}
int find_max(int now)
{
    if (!now) return 0;
    while (now)
    {
        if (a[now].right) now=a[now].right;
        else return now;
    }
}
bool del(int now)
{
    splay(now,0);
    if (!a[now].left&&!a[now].right){root=0;return 1;}
    else if (!a[now].left)
    {
        a[a[now].right].fa=0;
        root=a[now].right;
    }
    else if (!a[now].right)
    {
        a[a[now].left].fa=0;
        root=a[now].left;
    }
    else
    {
        int child=find_max(a[now].left);
        splay(child,root);
        root=child;
        a[child].right=a[now].right;
        a[a[now].right].fa=child;
        a[child].fa=0;
        ct(child);
    }
    return 1;
}
int find_next_max(int x)
{
    int now=root,k=0;
    while (now)
    {
        if (a[now].data>x)
            k=(a[k].data>a[now].data||!k)?now:k,
            now=a[now].left;
        else now=a[now].right;
    }
    if (k) splay(k,0);
    return k;
}
int find_next_min(int x)
{
    int now=root,k=0;
    while (now)
    {
        if (a[now].data<x)
            k=(a[k].data<a[now].data||!k)?now:k,
            now=a[now].right;
        else now=a[now].left;
    }
    if (k) splay(k,0);
    return k;
}
main()
{
    n=in();
    for (int i=1;i<=n;i++)
    {
        scanf("%d%d",&hand,&x);
        if (!root) flag=hand;
        if (flag==hand) insert(x);
        else
        {
            int p,p1=find_next_min(x),p2=find_next_max(x);
            if (p1&&(x-a[p1].data<=a[p2].data-x||!p2)) p=p1;
            else p=p2;
            ans=(ans+abs(a[p].data-x))%mod;
            del(p);
        }
    }
    printf("%d",ans);
}