BZOJ_1208_&_Codevs_1258_[HNOI2004]_宠物收养所_(平衡树/set)

描述

---

http://www.lydsy.com/JudgeOnline/problem.php?id=1208

(据说codevs要更新?就不放codevs的地址了吧...)

有宠物和人,每个单位都有一个与其他单位不同的值(正),如果现在有宠物,来了一个人,人取走和他的值最接近的宠物(如果有两个选值小的),如果现在有人,来了一个宠物,和它值最接近的人(如果有两个选值小的)取走它.求每次带宠物走两个单位的值的差的绝对值的总和.

1208: [HNOI2004]宠物收养所

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 6798  Solved: 2677
[Submit][Status][Discuss]

Description

最
近，阿Q开了一间宠物收养所。收养所提供两种服务：收养被主人遗弃的宠物和让新的主人领养这些宠物。每个领养者都希望领养到自己满意的宠物，阿Q根据领养
者的要求通过他自己发明的一个特殊的公式，得出该领养者希望领养的宠物的特点值a（a是一个正整数，a<2^31），而他也给每个处在收养所的宠物
一个特点值。这样他就能够很方便的处理整个领养宠物的过程了，宠物收养所总是会有两种情况发生：被遗弃的宠物过多或者是想要收养宠物的人太多，而宠物太
少。 1．
被遗弃的宠物过多时，假若到来一个领养者，这个领养者希望领养的宠物的特点值为a，那么它将会领养一只目前未被领养的宠物中特点值最接近a的一只宠物。
（任何两只宠物的特点值都不可能是相同的，任何两个领养者的希望领养宠物的特点值也不可能是一样的）如果有两只满足要求的宠物，即存在两只宠物他们的特点
值分别为a-b和a+b，那么领养者将会领养特点值为a-b的那只宠物。 2．
收养宠物的人过多，假若到来一只被收养的宠物，那么哪个领养者能够领养它呢？能够领养它的领养者，是那个希望被领养宠物的特点值最接近该宠物特点值的领养
者，如果该宠物的特点值为a，存在两个领养者他们希望领养宠物的特点值分别为a-b和a+b，那么特点值为a-b的那个领养者将成功领养该宠物。
一个领养者领养了一个特点值为a的宠物，而它本身希望领养的宠物的特点值为b，那么这个领养者的不满意程度为abs(a-b)。【任务描述】你得到了一年
当中，领养者和被收养宠物到来收养所的情况，希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。

Input

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

Output

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

Sample Input

5
0 2
0 4
1 3
1 2
1 5

Sample Output

3
(abs(3-2) + abs(2-4)=3，最后一个领养者没有宠物可以领养)

HINT

Source

分析

---

用平衡树查找前驱和后驱即可.依照题意,要么全是人要么全是宠物,所以建一棵树即可.模板题.

注意:

1.判断是否有前驱或后驱时要小心.

p.s.第一道用Splay写的题,调了好久...指针理解的不是很好啊.

Treap:

 #include <cstdio>
 #include <cstdlib>
 #include <algorithm>
 using namespace std;

 const int oo=~0u>>;

 struct Treap{
     struct node{
         node* ch[];
         int v,r,s;
         node(int v,node* t):v(v){ ch[]=ch[]=t; r=rand(); s=; }
         void push_up(){ s=ch[]->s+ch[]->s+; }
     }*root,*null;
     Treap(){
         null=new node(,NULL); null->s=; null->r=oo;
         root=null;
     }
     void rot(node* &o,bool d){
         node* k=o->ch[!d]; o->ch[!d]=k->ch[d]; k->ch[d]=o;
         o->push_up(); k->push_up(); o=k;
     }
     void insert(node* &o,int x){
         if(o==null) o=new node(x,null);
         else{
             bool d=x>o->v;
             insert(o->ch[d],x);
             if(o->ch[d]->r<o->r) rot(o,!d);
             o->push_up();
         }
     }
     void remove(node* &o,int x){
         if(o->v==x){
             if(o->ch[]!=null&&o->ch[]!=null){
                 bool d=o->ch[]->r<o->ch[]->r;
                 rot(o,d); remove(o->ch[d],x);
                 o->push_up();
             }
             else{
                 node* u=o;
                 if(o->ch[]==null) o=o->ch[]; else o=o->ch[];
                 delete u;
             }
         }
         else{
             bool d=x>o->v;
             remove(o->ch[d],x);
             o->push_up();
         }
     }
     int pre(int x){
         node* t=root;
         int ret=-;
         while(t!=null){
             if(t->v<x) ret=t->v,t=t->ch[];
             else t=t->ch[];
         }
         return ret;
     }
     int suc(int x){
         node* t=root;
         int ret=-;
         while(t!=null){
             if(t->v>x) ret=t->v,t=t->ch[];
             else t=t->ch[];
         }
         return ret;
     }
 }tree;

 const int maxn=+,mod=1e6;
 int n;
 int main(){
     scanf("%d",&n);
     int cnt=,now,ans=;
     while(n--){
         int a,b;
         scanf("%d%d",&a,&b);
         if(cnt==) now=a;
         if(a==now){
             tree.insert(tree.root,b);
             cnt++;
         }
         else{
             int pre=tree.pre(b);
             int suc=tree.suc(b);
             if(pre==-){
                 int    t=suc-b;
                 ans=(ans+t)%mod;
                 tree.remove(tree.root,suc);
             }
             else if(suc==-){
                 int t=b-pre;
                 ans=(ans+t)%mod;
                 tree.remove(tree.root,pre);
             }
             else{
                 int t1=b-pre,t2=suc-b;
                 if(t1<=t2){
                     ans=(ans+t1)%mod;
                     tree.remove(tree.root,pre);
                 }
                 else{
                     ans=(ans+t2)%mod;
                     tree.remove(tree.root,suc);
                 }
             }
             cnt--;
         }
     }
     printf("%d\n",ans);
     return ;
 }

 Treap

set:

 #include <bits/stdc++.h>
 using namespace std;

 const int mod=;
 int n,ans,a,b,now,cnt;
 set <int> st;
 int main(){
     scanf("%d",&n);
     set <int> :: iterator it;
     while(n--){
         scanf("%d%d",&a,&b);
         if(!cnt) now=a;
         if(a==now) st.insert(b), cnt++;
         else{
             it=st.lower_bound(b);
             if(it!=st.end()&&it!=st.begin()){
                 int suc=*it,pre=*(--it);
                 if(b-pre<=suc-b){
                     st.erase(pre);
                     ans=(ans+b-pre)%mod;
                 }
                 else{
                     st.erase(suc);
                     ans=(ans+suc-b)%mod;
                 }
             }
             else if(it!=st.end()){
                 int suc=*it;
                 st.erase(suc);
                 ans=(ans+suc-b)%mod;
             }
             else{
                 int pre=*(--it);
                 st.erase(pre);
                 ans=(ans+b-pre)%mod;
             }
             cnt--;
         }
     }
     printf("%d\n",ans);
     return ;
 }