BZOJ1208: [HNOI2004]宠物收养所

题目链接

因为宠物和领养人不可能同时存在，所以只用建一颗平衡树就可以了，记录下当前树中是谁。然后就是找前驱后继，删点插入。为了方便处理，所以先加入了极大值和极小值。

（真的是坑爸爸，0是正整数！！？）

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 80005 
#define M 200005
#define INF 0x7fffffff
#define mod 1000000
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,rt,cnt,now=-1,Pre,Next,ans;
int son[N][2],fa[N],sz[N],num[N];

void pushup(int k){
    sz[k]=sz[son[k][0]]+sz[son[k][1]]+1;
}

void Rotate(int x,int &k)
{
    int y=fa[x],z=fa[y],l,r;
    l=son[y][1]==x;r=l^1;
    if(y==k) k=x;
    else son[z][son[z][1]==y]=x;
    fa[x]=z;fa[y]=x;fa[son[x][r]]=y;
    son[y][l]=son[x][r];son[x][r]=y;
    pushup(y);pushup(x);
}

void Splay(int x,int &k)
{
    while(x!=k)
    {
        int y=fa[x],z=fa[y];
        if(y!=k) {
            if(son[z][0]==y^son[y][0]==x) Rotate(x,k);
            else Rotate(y,k);
        }
        Rotate(x,k);
    }
}

void Insert(int x,int k,int Last)
{
    if(!k)
    {
        k=++cnt;num[k]=x;fa[k]=Last;
        sz[k]=1;son[Last][x>num[Last]]=k;
        if(!rt) rt=k;else Splay(k,rt);return;
    }
    if(num[k]<x) Insert(x,son[k][1],k);
    else Insert(x,son[k][0],k);
}

void Get_Pre(int x,int k)
{
    if(!k) return;
    if(num[k]<=x)
    {
        Pre=k;
        Get_Pre(x,son[k][1]);
    }
    else Get_Pre(x,son[k][0]);
}

void Get_Next(int x,int k)
{
    if(!k) return;
    if(num[k]>=x)
    {
        Next=k;
        Get_Next(x,son[k][0]);
    }
    else Get_Next(x,son[k][1]);
}

int Find(int k,int x)
{
    int l=son[k][0],r=son[k][1];
    if(!k) return 0;
    if(sz[l]+1==x) return k;
    if(sz[l]+1>x) return Find(l,x);
    return Find(r,x-sz[l]-1);
}

void Del(int k)
{
    int x=Find(rt,k-1),y=Find(rt,k+1);
    Splay(x,rt);Splay(y,son[x][1]);
    int z=son[y][0];son[y][0]=0,fa[z]=sz[z]=0;
    pushup(y);pushup(x);
}

void Solve(int x)
{
    int tmp,k=0;
    Get_Pre(x,rt);Get_Next(x,rt);
    if(Pre!=1) tmp=x-num[Pre],k=Pre;
    if(Next!=2&&(Pre==1||num[Next]-x<x-num[Pre])) tmp=num[Next]-x,k=Next;
    ans+=tmp;while(ans>=mod) ans-=mod;
    Splay(k,rt);k=sz[son[k][0]]+1;
    Del(k);if(sz[rt]==2) now=-1;
}

void print(int x)
{
    if(son[x][0]) print(son[x][0]);
    printf("%d\n",x);
    if(son[x][1]) print(son[x][1]);
}

int main()
{
    n=read();Insert(-1,rt,0);
    Insert(INF,rt,0);
    for(int i=1;i<=n;i++)
    {
        static int x,y;
        x=read(),y=read();
        if(now!=x) Insert(y,rt,0),now=x^1;
        else Solve(y);
    }
    printf("%d\n",ans);
    return 0;
}