bzoj1208: [HNOI2004]宠物收养所

应某人要求把这题放出来QAQ。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,tree[80005][2],root,t1,t2,size,ans,fa[80005],s[80005],x1,x2;
int mod=1000000;
int read()
{
    char ch=getchar();int f=0;
    while(ch<'0'||ch>'9')
    ch=getchar();
    while(ch>='0'&&ch<='9')
    {f=(f<<1)+(f<<3)+ch-'0';
    ch=getchar();
    }
    return f;
}
void rotate(int x,int &k)
{
    int y=fa[x],z=fa[y],l,r;
    if(tree[y][0]==x) l=0;else l=1;
    r=l^1;
    if(y==k)
    k=x;
    else
    {
        if(tree[z][0]==y)
        {
            tree[z][0]=x;
        }
        else tree[z][1]=x;
    }
    fa[x]=z;fa[y]=x;fa[tree[x][r]]=y;
    tree[y][l]=tree[x][r];tree[x][r]=y;
}
void splay(int x,int &k)
{
    int y,z;
    while(x!=k)
    {
        y=fa[x],z=fa[y];
        if(y!=k)
        {
            if((tree[z][0]==y)^(tree[y][0]==x)) rotate(x,k);
            else rotate(y,k);
        }
        rotate(x,k);
    }
}

void del(int x)
{
    splay(x,root);
    if(!tree[x][0]) root=tree[x][1];
    else if(!tree[x][1]) root=tree[x][0];
    else
    {
        int k=tree[x][1];
        while(tree[k][0])
        k=tree[k][0];
        tree[k][0]=tree[x][0];fa[tree[x][0]]=k;
        root=tree[x][1];
    }
    fa[root]=0;
    return;
}
void before(int k,int x)
{
    if(!k) return;
    if(x>=s[k])
    {
        t1=s[k];x1=k;before(tree[k][1],x);return;
    }
    else
    before(tree[k][0],x);
}
void after(int k,int x)
{
    if(!k)
    return;
    if(x<=s[k])
    {
        t2=s[k];x2=k;after(tree[k][0],x);return;
    }
    else after(tree[k][1],x);
}
void add(int &k,int x,int f)
{
    if(!k)
    {
        k=++size;
        s[size]=x;
        fa[k]=f;
        splay(k,root);
        return;
    }
    if(x<s[k]) add(tree[k][0],x,k);
    else add(tree[k][1],x,k);
}
int main()
{
    int x,k,last;
    n=read();
    for(int i=1;i<=n;i++)
    {
        k=read(),x=read();x1=x2=0;
        if(!root||last==k) {last=k;add(root,x,0);continue;}
        before(root,x);after(root,x);
        if((!x2||x-t1<=t2-x)&&x1) ans=(ans+x-t1)%mod,del(x1);
        else if(x2&&(!x1||t2-x<x-t1)) ans=(ans+t2-x)%mod,del(x2);
    }
    cout<<ans;
}