HNOI2004宠物收养所

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=1208

 

 

1208: [HNOI2004]宠物收养所

Time Limit: 10 Sec  Memory Limit: 162 MB

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

Splay

做法：裸Splay或者直接用Set（红黑树啧啧啧。。用lower_bound-1和upper_bound检测下即可)

Codes:

#include <set>
#include <queue>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int Mod = 1000000;
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,l,r) for(int i=l;i<=r;i++)
set<int>::iterator Min,Max;
set<int> T;
int n,nowa,nowb,ans;

int main(){
    scanf("%d",&n);
    For(i,n){int op,Val;
        scanf("%d %d",&op,&Val);
        if(op){
            if(nowa){
                nowa--;
                Min = T.lower_bound(Val);
                if((*Min!=Val)&&(Min!=T.begin())) 
                    Min--;
                Max = T.upper_bound(Val);
                if(abs(*Min-Val)<=abs(*Max-Val)){
                    ans = (ans + abs(*Min-Val)) % Mod;
                    T.erase(Min);
                }
                else{
                    ans = (ans + abs(*Max-Val)) % Mod;
                    T.erase(Max);
                }
            }
            else{
                T.insert(Val);
                nowb++;    
            }
        }else{
            if(nowb){
                nowb--;          
                Min = T.lower_bound(Val);
                if((*Min!=Val)&&(Min!=T.begin())) Min--;
                Max = T.upper_bound(Val);
                if(abs(*Min-Val)<=abs(*Max-Val)){
                    ans = (ans + abs(*Min-Val)) % Mod;
                    T.erase(Min);
                }
                else{
                    ans = (ans + abs(*Max-Val)) % Mod;
                    T.erase(Max);
                }
            }else{
                nowa++;
                T.insert(Val);
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}
----------------------下面是我的渣渣splay..-------------------- 
#include <set>
#include <cmath>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 80010;
const int Mod = 1000000;
const int maxint = 2147483647;
#define fa(i) (T[i].p)
#define Loc(i) (T[fa(i)].s[1]==i)
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,l,r) for(int i=l;i<=r;i++)
#define Sets(a,b,Loc) {T[a].s[Loc] = b;fa(b) = a;}
 
struct tnode{
    int s[2],v,p;
}T[N];
 
int n,root,nowa,nowb,tot;
int ans;
 
void Rot(int i){
    int x = i , y = fa(x) , z = fa(y);
    int lx = Loc(x) , ly = Loc(y) , lz = Loc(z);
    Sets(y,T[x].s[!lx],lx);
    Sets(z,x,ly);
    Sets(x,y,!lx);
}
 
void Splay(int i,int goal){
    for(;fa(i)!=goal;){
        if(fa(fa(i))!=goal) Rot(fa(i));
        Rot(i);
    }
    if(!goal) root = i;
}
 
int Pred(int i){
    int Next = T[i].s[0];
    if(!Next) return Next;
    while(T[Next].s[1]) Next = T[Next].s[1];
    return Next;
}
 
int Succ(int i){
    int Next = T[i].s[1]; 
    if(!Next) return Next;
    while(T[Next].s[0]) Next = T[Next].s[0];
    return Next;
}
 
void Delete(int i){
    Splay(i,0);
    if(T[i].s[0]){
        Splay(Pred(i),i);
        Sets(T[i].s[0],T[i].s[1],1);
        fa(T[i].s[0]) = 0;
        root = T[i].s[0];
    }else{
        if(T[i].s[1]){root = T[i].s[1];fa(T[i].s[1]) = 0;}
        else          root = 0;
    }
    T[i].s[0] = T[i].s[1] = T[i].v = T[i].p = 0;
}
 
int Find(int V,int i){
    if(V==T[i].v){
        Delete(i);
        return V;
    }else
    if(V>T[i].v){
        Splay(i,0);
        int succ = Succ(i);
        if((V>T[succ].v)&&(succ)) return Find(V,succ);
        else{
            if(abs(V-T[i].v)<=abs(V-T[succ].v)){
                int temp = T[i].v;
                Delete(i);
                return temp;
            }else{
                int temp = T[succ].v;
                Delete(succ);
                return temp;
            }
        }
    }else{
        Splay(i,0);
        int pred = Pred(i);
        if((V<T[pred].v)&&(pred)) return Find(V,pred);
        else{
            if(abs(V-T[pred].v)<=abs(V-T[i].v)){
                int temp = T[pred].v;                
                Delete(pred);
                return temp;
            }else{
                int temp = T[i].v;
                Delete(i);
                return temp;
            }
        }
    } 
}
 
void Insert(int V,int i){
    if(!i){
        root = ++tot;T[root].v = V;T[root].p = 0;
        return;
    }
    if(!T[i].s[V>T[i].v]){
        T[i].s[V>T[i].v] = ++tot;T[tot].p = i;T[tot].v = V;
        Splay(tot,0);
    }else Insert(V,T[i].s[V>T[i].v]);
}
 
int main(){
    //freopen("pet.in","r",stdin);
    //freopen("output.out","w",stdout);
    scanf("%d",&n);T[0].v = maxint;
    For(i,n){int a,b;
        scanf("%d%d",&a,&b);
        if(a){
            if(nowa){
                nowa--;
                ans = ( ans % Mod + abs(Find(b,root) - b) % Mod ) % Mod;
            }
            else {
                Insert(b,root);
                nowb++;
            }
        }else{
            if(nowb){
                nowb--;
                ans = ( ans % Mod + abs(Find(b,root) - b) % Mod ) % Mod;
            }
            else{
                Insert(b,root);
                nowa++;
            }
        }
    }
    printf("%d\n",ans%Mod);
    return 0;
}