宠物收养场 Treap

Posted 2020-09-21 kakakakakaka

宠物收养场

时间限制: 1 Sec  内存限制: 128 MB

题目描述

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

每个领养者都希望领养到自己满意的宠物，凡凡根据领养者的要求通过他自己发明的一个特殊的公式，得出该领养者希望领养的宠物的特点值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)。

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

 

输入

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

 

输出

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

 

样例输入

5

0 2

0 4

1 3

1 2

1 5

样例输出

3

提示

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

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

数据有一定梯度

对于80%的数据，收养者按照到来的顺序收养宠物

 

题解：

刚看到这题，果断treap。

仔细一想发现可以不用两个treap，因为树里面要么全是宠物要么全是人，所以只需要一个就够了。（关键是看到无论是人还是宠物，查找的条件是一样的，即：”存在两只宠物他们的特点值分别为a-b和a+b，那么领养者将会领养特点值为a-b的那只宠物“and”存在两个领养者他们希望领养宠物的特点值分别为a-b和a+b，那么特点值为a-b的那个领养者将成功领养该宠物“所以此题必有蹊跷，心机吧~）

然后就简单了，不过查找的时候应该找一遍前驱再找一遍后继，不能一起找，否则会出错。

嗯~就是这样。

AC代码（注意取模）：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<stack>
#include<ctime>
#include<vector>
using namespace std;
int n,m,l=3;
struct node
{
    int key,rev;
    node *child[2];
} bst[20001],*root=NULL,*pos=bst;
queue<node*>mem;
void rotate(node* &r,bool t)
{
    node *y=r->child[!t];
    r->child[!t]=y->child[t];
    y->child[t]=r;
    r=y;
}
void newnode(node* &r,int key)
{
    if(!mem.empty())r=mem.front(),mem.pop();
    else r=pos++;
    r->key=key;
    r->rev=rand();
    r->child[0]=r->child[1]=NULL;
}
void insert(node* &r,int key)
{
    if(!r)newnode(r,key);
    else
    {
        bool t=r->key<key;
        insert(r->child[t],key);
        if(r->rev>r->child[t]->rev)rotate(r,!t);
    }
}
void delet(node* &r,int key)
{
    if(!r)return;
    if(r->key==key)
    {
        if(r->child[1]&&r->child[0])
        {
            bool t=r->child[0]->rev<r->child[1]->rev;
            rotate(r,t);
            delet(r->child[t],key);
        }
        else
        {
            mem.push(r);
            if(r->child[0])r=r->child[0];
            else r=r->child[1];
        }
    }
    else delet(r->child[r->key<key],key);
}
int pre,nxt;
void getspre(node* r,int key)
{
    if(r==NULL)return;
    if(r->key>key)getspre(r->child[0],key);
    else pre=r->key,getspre(r->child[1],key);
}
void getsnext(node* r,int key)
{
    if(r==NULL)return;
    if(r->key<key)getsnext(r->child[1],key);
    else nxt=r->key,getsnext(r->child[0],key);
}
int main()
{
    int i,j,k,ans=0;
    scanf("%d",&n);
    for(i=1; i<=n; i++)
    {
        scanf("%d%d",&j,&k);
        if(j==l||root==NULL)
        {
            insert(root,k);
            l=j;
        }
        else if(j!=l)
        {
            pre=nxt=2000000000;
            getspre(root,k);
            getsnext(root,k);
            if(abs(k-pre)>abs(k-nxt))
            {
                ans+=abs(k-nxt);
                ans%=1000000;
                delet(root,nxt);
            }
            else
            {
                ans+=abs(k-pre);
                ans%=1000000;
                delet(root,pre);
            }
        }
        if(root==NULL)l=3;
    }
    cout<<ans;
    return 0;
}