[HNOI2004]宠物收养所

题目链接

[HNOI2004]宠物收养所

题解

这基本上就是一个裸的treap，听说set可以秒此题，但是本着练习treap的思想我还是手打了一份平衡树。首先人和宠物是一样的，只需要判断一下现在收养所里面装的是什么，来东西了，一样就直接insert，不一样就找一下前驱后继判一下选哪个然后把那个remove了，每次判一下树是否为空，空了就换成当前来的东西的类型。应该比较简单codevs上都是钻石（我猜是因为set可以做就很简单了）

代码

我是拿的板子所以多写了rank和kth

#include<queue>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
struct Node{
    int v,r,s;
    Node *ch[2];
    Node(int v):v(v){r=rand();s=1;ch[0]=ch[1]=NULL;}
    int cmp(int x){
        if(v==x) return -1;
        return v>x?0:1;
    }
    void maintain(){
        s=1;
        if(ch[0]!=NULL) s+=ch[0]->s;
        if(ch[1]!=NULL) s+=ch[1]->s;
    }
};

void rotate(Node* &p,int c)
{
    Node *t=p->ch[c^1];
    p->ch[c^1]=t->ch[c];
    t->ch[c]=p;
    p->maintain();
    t->maintain();p=t;
    return ;
}

void insert(Node* &p,int x)
{
    if(p==NULL){p=new Node(x);return ;}
    int tcmp=p->cmp(x);
    //int tcmpa=(x>p->v ? 1:0) 有重复的时候用这一句 
    insert(p->ch[tcmp],x);
    if(p->ch[tcmp]->r > p->r) rotate(p,tcmp^1);
    p->maintain();
}

void remove(Node* &p,const int x)
{
    int tcmp=p->cmp(x); 
    if(tcmp==-1)
    {
        if(p->ch[0]==NULL) p=p->ch[1];//**remove(p->ch[1],x)
        else if(p->ch[1]==NULL) p=p->ch[0];//**remove(p->ch[0],x)
        else{
            int tmp=(p->ch[0]->r > p->ch[1]->r ? 1:0);
            rotate(p,tmp);remove(p->ch[tmp],x);//**p=p->ch[tmp];
        }
    }
    else remove(p->ch[tcmp],x);
    if(p!=NULL) p->maintain();//** 漏了一句 
    return ;
}

bool find(Node* p,const int x)
{
    while(p!=NULL)
    {
        int tcmp=p->cmp(x);
        if(tcmp==-1) return true;
        p=p->ch[tcmp];
    }
    return false;
}

int kth(Node* p,int x)
{
    if(p==NULL||x<0||x>p->s)return -1;
    int d=(p->ch[1]==NULL?0:p->ch[1]->s);
    if(x==d+1) return p->v;
    if(x<=d) return kth(p->ch[1],x);
    return kth(p->ch[0],x-d-1);
} 

int rank(Node* p,const int x)
{
    int ret=1;
    while(p!=NULL)
    {
        int tcmp=p->cmp(x),d=(p->ch[1]==NULL?0:p->ch[1]->s);
        if(tcmp==-1) return ret+d;
        if(tcmp==0) ret=ret+1+d;
        else ret++;
        p=p->ch[tcmp];
    }
    return ret;
}

int pre(Node* p,const int x)
{
    int ret=0;
    while(p!=NULL)
    {
        int tcmp=p->cmp(x);
        if(tcmp==-1) return x;
        if(tcmp) ret=max(ret,p->v);
        p=p->ch[tcmp];
    }
    return ret;
}

const int INF=(1<<30);
int last(Node* p,const int x)
{
    int ret=INF;
    while(p!=NULL)
    {
        int tcmp=p->cmp(x);
        if(tcmp==-1) return x;
        if(!tcmp) ret=min(ret,p->v);
        p=p->ch[tcmp];
    }
    return ret;
}

inline int read()
{
    int ret=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    for(;ch>='0'&&ch<='9';ch=getchar()) ret=ret*10+ch-'0';
    return ret;
}

const int MOD=1000000;
int main()
{
    srand(0);
    Node *root=NULL;
    int kind,n,sum=0;cin>>n;
    for(int i=1,ins,x;i<=n;i++)
    {
        ins=read();x=read();
        if(root==NULL) kind=ins,insert(root,x);
        else if(ins==kind) insert(root,x);
        else{
            int PRE=pre(root,x);
            int LAST=last(root,x);
            int d=INF,wh;
            if(PRE&&abs(PRE-x)<d) d=abs(PRE-x),wh=PRE;
            if(LAST!=INF&&abs(LAST-x)<d) d=abs(LAST-x),wh=LAST;
            sum=(sum+d)%MOD;
            remove(root,wh);
        }
    }
    cout<<sum;
    return 0;
}