首页 > 分享 > [BZOJ1208]宠物收养所（Splay）

[BZOJ1208]宠物收养所（Splay）

萌宠菠菠乐园
2024-10-21 18:44

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_PREelse{if(k[nex]-t<t-k[pre]) Get_NEXelse Get_PRE}}}printf("%dn",Ans);return 0; }