洛谷P2286 [HNOI2004]宠物收养场【Treap】题解+AC代码

题目传送门啦~啦~啦~

题目描述

凡凡开了一间宠物收养场。收养场提供两种服务：收养被主人遗弃的宠物和让新的主人领养这些宠物。

每个领养者都希望领养到自己满意的宠物，凡凡根据领养者的要求通过他自己发明的一个特殊的公式，得出该领养者希望领养的宠物的特点值a（a是一个正整数，a<2^31），而他也给每个处在收养场的宠物一个特点值。这样他就能够很方便的处理整个领养宠物的过程了，宠物收养场总是会有两种情况发生：被遗弃的宠物过多或者是想要收养宠物的人太多，而宠物太少。

被遗弃的宠物过多时，假若到来一个领养者，这个领养者希望领养的宠物的特点值为a，那么它将会领养一只目前未被领养的宠物中特点值最接近a的一只宠物。（任何两只宠物的特点值都不可能是相同的，任何两个领养者的希望领养宠物的特点值也不可能是一样的）如果有两只满足要求的宠物，即存在两只宠物他们的特点值分别为a-b和a+b，那么领养者将会领养特点值为a-b的那只宠物。

收养宠物的人过多，假若到来一只被收养的宠物，那么哪个领养者能够领养它呢？能够领养它的领养者，是那个希望被领养宠物的特点值最接近该宠物特点值的领养者，如果该宠物的特点值为a，存在两个领养者他们希望领养宠物的特点值分别为a-b和a+b，那么特点值为a-b的那个领养者将成功领养该宠物。

一个领养者领养了一个特点值为a的宠物，而它本身希望领养的宠物的特点值为b，那么这个领养者的不满意程度为abs(a-b)。

你得到了一年当中，领养者和被收养宠物到来收养所的情况，请你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。

输入格式：

第一行为一个正整数n，n<=80000，表示一年当中来到收养场的宠物和领养者的总数。接下来的n行，按到来时间的先后顺序描述了一年当中来到收养场的宠物和领养者的情况。每行有两个正整数a, b，其中a=0表示宠物，a=1表示领养者，b表示宠物的特点值或是领养者希望领养宠物的特点值。（同一时间呆在收养所中的，要么全是宠物，要么全是领养者，这些宠物和领养者的个数不会超过10000个）

输出格式：

仅有一个正整数，表示一年当中所有收养了宠物的领养者的不满意程度的总和mod 1000000以后的结果。

输入样例

5

0 2

0 4

1 3

1 2

1 5

输出样例

3

注：abs(3-2) + abs(2-4)=3，

最后一个领养者没有宠物可以领养。

---

题目分析：

主要就是要建两棵树

一棵是顾客，一棵是宠物

如果来了宠物且顾客的树不为空，就按题目要求找顾客

顾客来了也是同理

剩下的就是Treap的基本操作啦

如果有不懂得小伙伴可以戳这里

其他注释具体见代码

---

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
#include<queue>
using namespace std;

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int inf=1e9;
int n;
int key,ans;
struct node
{
    node* ch[2];
    int v,r,cnt;

    node(int v) :v(v){ cnt=1; r=rand(); ch[0]=ch[1]=NULL;}
    int cmp(int x) const{ if(x==v) return -1; return x<v ?0:1; }
};
node* rtpet=NULL;//记录宠物的Treap
node* rtper=NULL;//记录顾客

void rotate(node* &p,int d)
{
    node* k=p->ch[d^1];
    p->ch[d^1]=k->ch[d];
    k->ch[d]=p;
    p=k;
}

void ins(node* &p,int x)
{
    if(p==NULL){ p=new node(x); return; }
    int d=p->cmp(x);
    if(d==-1){p->cnt++;return;}
    ins(p->ch[d],x);
    if( p->ch[d]->r < p->r )rotate(p,d^1);
}

void del(node* &p,int x)
{
    if(p==NULL) return;
    if(x==p->v)
    {
        if(p->cnt>1){p->cnt--;return;}
        else
        {
            if(p->ch[0]==NULL)
            { node* k=p; p=p->ch[1]; delete(k); return;}
            else if(p->ch[1]==NULL)
            { node* k=p; p=p->ch[0]; delete(k); return;}

            else
            {
                int dd=p->ch[0]->r < p->ch[1]->r ?1:0;
                rotate(p,dd); del(p->ch[dd],x);
                return;
            }
        }
    }
    else if(x<p->v) del(p->ch[0],x);
    else del(p->ch[1],x);
}

int get(node* p,int x)//这里我们返回的值是最小不满意度
{
    if(p==NULL) return inf;//如果结点为空，返回inf确保不影响查找
    int d=p->cmp(x);
    if(d==-1){ key=p->v;return 0;}
    //如果过正好找到符合的，令key记录待删除结点的权值，并返回不满意度为0
    else
    {
        //没有正好符合就继续递归寻找
        int minn=get(p->ch[d],x);
        if( abs(p->v-x) < minn )
        {
            key=p->v;
            return abs(p->v-x);
        }//比较当前结点和其孩子的不满意度
        else
        return minn;
    }
}

int main()
{
    n=read();
    while(n--)
    {
        int x=read(),y=read();
        if(x==0)
        {
            //如果顾客不为空就查找，反之则插入宠物的Treap
            if(rtper!=NULL)
            {
                int d=get(rtper,y);
                ans+=d;ans%=1000000;
                del(rtper,key);//找到后要删除该顾客
            }
            else ins(rtpet,y);
        }
        else if(x==1)
        {
            //操作同上，就是反过来
            if(rtpet!=NULL)
            {
                int d=get(rtpet,y);
                ans+=d;ans%=1000000;
                del(rtpet,key);
            }
            else ins(rtper,y);
        }
    }
    cout<<ans%1000000;//一定要记得取余！！！为这个查了好久QAQ
    return 0;
}