Bzoj1208宠物收养所(Splay)

Posted void_f

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Bzoj1208宠物收养所(Splay)相关的知识,希望对你有一定的参考价值。

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)。

【任务描述】你得到了一年当中,领养者和被收养宠物到来收养所的情况,希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时,收养所里面既没有宠物,也没有领养者。

Solution

差不多是splay模板题,细节打错调了好久(——)

刚开始看以为要建2棵平衡树,其实一个就行,只要记录当前储存的是宠物还是领养者

需要操作:插入,删除

!!插入得考虑树为空的情况,WA了好久

删除一个节点X:先把X旋到根,然后让后驱代替X即可

Code

#include <cstdio>
#include <algorithm>
#include <cmath>
#define N 80010
#define Get_NEX {Ans+=k[nex]-t;Ans%=mod;Del(nex);}
#define Get_PRE {Ans+=t-k[pre];Ans%=mod;Del(pre);}
#define lc(x) T[(x)][0]
using namespace std;

const int mod=1e6;
int n,kind,rt,T[N][2],tot,k[N],fa[N],Ans,pre,nex;

inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void rotate(int p){
    int q=fa[p],y=fa[q],x=(T[q][1]==p);
    T[q][x]=T[p][x^1];fa[T[q][x]]=q;
    T[p][x^1]=q;fa[q]=p;
    fa[p]=y;
    if(y){
        if(T[y][0]==q) T[y][0]=p;
        else if(T[y][1]==q) T[y][1]=p;
    }
}

inline void splay(int x){
    for(int y;y=fa[x];rotate(x))
        if(fa[y]) rotate((x==lc(y))==(y==lc(fa[y]))?y:x);
    rt=x;
}

void Insert(int x,int v){
    if(!rt){
        rt=++tot;
        T[rt][0]=T[rt][1]=fa[rt]=0;
        k[rt]=v;
        return;
    }
    
    int y;
    while(1){
        y=T[x][k[x]<v];
        if(!y){
            y=++tot;
            k[y]=v;
            T[y][0]=T[y][1]=0;
            fa[y]=x;
            T[x][k[x]<v]=y;
            break;
        }
        x=y;
    }
    splay(y);
}

void Pre(int u,int x){
    if(u==0) return;
    if(k[u]<=x){pre=u;Pre(T[u][1],x);}
    else Pre(T[u][0],x);
}

void Nex(int u,int x){
    if(u==0) return;
    if(k[u]>=x){nex=u;Nex(T[u][0],x);}
    else Nex(T[u][1],x);
}

void Del(int x){
    splay(x);
    if(T[x][0]*T[x][1]==0) rt=T[x][0]+T[x][1];
    else{
        int t=T[x][1];
        while(T[t][0]) t=T[t][0];
        T[t][0]=T[x][0];fa[T[x][0]]=t;
        rt=T[x][1];
    }
    fa[rt]=0;
}

int main(){
    n=read();
    while(n--){
        int op=read(),t=read();
        if(!rt) {kind=op;Insert(rt,t);}
        else if(kind==op) Insert(rt,t);
        else{
            pre=-1,nex=-1;
            Pre(rt,t),Nex(rt,t);
            if(pre==-1) Get_NEX else if(nex==-1) Get_PRE
            else{
                if(k[nex]-t<t-k[pre]) Get_NEX 
                else Get_PRE
            }
        }
    }
    printf("%d\n",Ans);
    return 0;
} 

以上是关于Bzoj1208宠物收养所(Splay)的主要内容,如果未能解决你的问题,请参考以下文章

BZOJ1208宠物收养所(平衡树,splay)

BZOJ1208 [HNOI2004]宠物收养所

BZOJ1208: [HNOI2004]宠物收养所

AC日记——宠物收养所 bzoj 1208

1208. [HNOI2004]宠物收养场平衡树-splay

BZOJ1208: [HNOI2004]宠物收养所