[bzoj1208][HNOI2004]宠物收养所 SBT删点,前驱,后继

1208: [HNOI2004]宠物收养所

Time Limit:

最近,阿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,最后一个领养者没有宠物可以领养)

数据不明,不在codevs上测还不知道n有1000000,最大有千万级。
关键是我为什么如此慢,倒数的,STL怕是比我快啊

#include<cstdio>
const int mod = 1000000;
struct SBT{
    int son[2],cnt,v,size;
}e[1000005];
int n,cnt,root,ans=0,pre,suc;
int Abs( int X ){
    if(X < 0) return (-X);
    else return X;
}
inline void rotate( int &k, bool f ){
    int p = e[k].son[f]; e[k].son[f] = e[p].son[!f]; e[p].son[!f] = k;
    e[k].size = e[e[k].son[1]].size + e[e[k].son[2]].size + e[k].cnt;
    e[p].size = e[e[p].son[1]].size + e[e[p].son[2]].size + e[p].cnt; 
    k = p;
}
void maintain( int &k, bool f ){
    if(!k) return; 
    if(e[e[k].son[!f]].size < e[e[e[k].son[f]].son[f]].size) 
        rotate(k,f);
    else if( e[e[k].son[!f]].size < e[e[e[k].son[f]].son[!f]].size ){
        rotate(e[k].son[f],!f);
        rotate(k,f);
    }
    else return;
    maintain(e[k].son[f],f);
    maintain(k,f);
}
void insert( int &k, int x ){
    if(!k){ 
        k = ++cnt; 
        e[k].size = e[k].cnt = 1; 
        e[k].v = x; 
        return ; 
    }
    e[k].size++; 
    if( e[k].v == x ) { 
        e[k].cnt++; 
        return ; 
    }
    insert(e[k].son[e[k].v<x],x); 
    maintain(k,e[k].v<x);
}
int del( int &k, int x ){
    if(!k) return k; 
    int tmp;
    if( e[k].v == x ){
        if( e[k].cnt > 1 ){ 
            e[k].size--; e[k].cnt--; 
            return k; 
        }else if( !(e[k].son[0]*e[k].son[1]) ){ 
            k = e[k].son[0] + e[k].son[1]; 
        }else { 
            tmp = del(e[k].son[0],x+1); 
            e[k].cnt = e[tmp].cnt; 
            e[k].v = e[tmp].v; 
            e[k].size = e[e[k].son[1]].size + e[e[k].son[0]].size + e[k].cnt; 
            return k;
        }
    } else if( (e[k].v<x && !e[k].son[1])||(e[k].v>x && !e[k].son[0]) ){
        tmp=k;k=e[k].son[0];
        e[k].size = e[e[k].son[0]].size + e[e[k].son[1]].size + e[k].cnt;
        return tmp;
    }
    else tmp = del(e[k].son[e[k].v<x],x);
    e[k].size = e[e[k].son[0]].size + e[e[k].son[1]].size + e[k].cnt; 
    return tmp;
}
void pred( int k, int x ){
    if( k == 0 ) return ;
    if( e[k].v <= x ){ 
        pre = k; 
        pred(e[k].son[1],x); 
    }
    else pred(e[k].son[0],x);
}
void succ( int k, int x ){
    if( k == 0 ) return ;
    if( e[k].v > x ){ 
        suc = k; 
        succ(e[k].son[0],x); 
    }
    else succ(e[k].son[1],x);
}
template<class T>inline void read(T &res){
    static char ch;T flag=1;
    while((ch=getchar())>'9'||ch<'0')if(ch=='-')flag=-1;res=ch-48;
    while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
int main(){
    read(n); int fir;
    for( int i = 1,a,b; i <= n; i++ ){
        read(a);read(b);
        if( e[root].size == 0 || a == fir ){
            fir = a;
            insert(root,b);
            continue;
        }
        pre = suc = 0;
        pred(root,b); succ(root,b);
        pre = (pre==0)?1000000000:e[pre].v;
        suc = (suc==0)?1000000000:e[suc].v;
        if(Abs(pre-b)<=Abs(suc-b)){del(root,pre);ans+=Abs(b-pre)%mod;}
        else{del(root,suc);ans+=Abs(b-suc)%mod;}
        ans%=mod;
    }
    printf("%d", ans);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值