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>>1;

struct Treap{
    struct node{
        node* ch[2];
        int v,r,s;
        node(int v,node* t):v(v){ ch[0]=ch[1]=t; r=rand(); s=1; }
        void push_up(){ s=ch[0]->s+ch[1]->s+1; }
    }*root,*null;
    Treap(){
        null=new node(0,NULL); null->s=0; 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[0]!=null&&o->ch[1]!=null){
                bool d=o->ch[0]->r<o->ch[1]->r;
                rot(o,d); remove(o->ch[d],x);
                o->push_up();
            }
            else{
                node* u=o;
                if(o->ch[0]==null) o=o->ch[1]; else o=o->ch[0];
                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=-1;
        while(t!=null){
            if(t->v<x) ret=t->v,t=t->ch[1];
            else t=t->ch[0];
        }
        return ret;
    }
    int suc(int x){
        node* t=root;
        int ret=-1;
        while(t!=null){
            if(t->v>x) ret=t->v,t=t->ch[0];
            else t=t->ch[1];
        }
        return ret;
    }
}tree;

const int maxn=80000+5,mod=1e6;
int n;
int main(){
    scanf("%d",&n);
    int cnt=0,now,ans=0;
    while(n--){
        int a,b;
        scanf("%d%d",&a,&b);
        if(cnt==0) now=a;
        if(a==now){
            tree.insert(tree.root,b);
            cnt++;
        }
        else{
            int pre=tree.pre(b);
            int suc=tree.suc(b);
            if(pre==-1){
                int    t=suc-b;
                ans=(ans+t)%mod;
                tree.remove(tree.root,suc);
            }
            else if(suc==-1){
                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
",ans);
    return 0;
}

Treap

set:

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

const int mod=1000000;
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
",ans);
    return 0;
}