题目描述
凡凡开了一间宠物收养场。收养场提供两种服务:收养被主人遗弃的宠物和让新的主人领养这些宠物。
每个领养者都希望领养到自己满意的宠物,凡凡根据领养者的要求通过他自己发明的一个特殊的公式,得出该领养者希望领养的宠物的特点值a(a是一个正整数,a<2^31),而他也给每个处在收养场的宠物一个特点值。这样他就能够很方便的处理整个领养宠物的过程了,宠物收养场总是会有两种情况发生:被遗弃的宠物过多或者是想要收养宠物的人太多,而宠物太少。
被遗弃的宠物过多时,假若到来一个领养者,这个领养者希望领养的宠物的特点值为a,那么它将会领养一只目前未被领养的宠物中特点值最接近a的一只宠物。(任何两只宠物的特点值都不可能是相同的,任何两个领养者的希望领养宠物的特点值也不可能是一样的)如果有两只满足要求的宠物,即存在两只宠物他们的特点值分别为a-b和a+b,那么领养者将会领养特点值为a-b的那只宠物。
收养宠物的人过多,假若到来一只被收养的宠物,那么哪个领养者能够领养它呢?能够领养它的领养者,是那个希望被领养宠物的特点值最接近该宠物特点值的领养者,如果该宠物的特点值为a,存在两个领养者他们希望领养宠物的特点值分别为a-b和a+b,那么特点值为a-b的那个领养者将成功领养该宠物。
一个领养者领养了一个特点值为a的宠物,而它本身希望领养的宠物的特点值为b,那么这个领养者的不满意程度为abs(a-b)。
你得到了一年当中,领养者和被收养宠物到来收养所的情况,请你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时,收养所里面既没有宠物,也没有领养者。
输入格式
第一行为一个正整数n,n<=80000,表示一年当中来到收养场的宠物和领养者的总数。接下来的n行,按到来时间的先后顺序描述了一年当中来到收养场的宠物和领养者的情况。每行有两个正整数a, b,其中a=0表示宠物,a=1表示领养者,b表示宠物的特点值或是领养者希望领养宠物的特点值。(同一时间呆在收养所中的,要么全是宠物,要么全是领养者,这些宠物和领养者的个数不会超过10000个)
输出格式
仅有一个正整数,表示一年当中所有收养了宠物的领养者的不满意程度的总和mod 1000000以后的结果。
输入输出样例
输入 #1复制
5
0 2
0 4
1 3
1 2
1 5
输出 #1复制
3
注:abs(3-2) + abs(2-4)=3,
最后一个领养者没有宠物可以领养。
敲了半天splay忽然发现其实set就行了。。。。。。(离散化之后,树状数组也行,只不过多个log)
这道题我们其实没必要用两颗splay,因为树上要么只有人,要么只有宠物,所以我们用变量计算一下当前人多还是宠物多即可。
注意计算前驱后继时应该是非严格前驱和非严格后继。
敲敲splay就当练手速了。。
AC代码:
#pragma GCC optimize(2)
#include<bits/stdc++.h>
//#define int long long
using namespace std;
const int N=1e5+10,inf=0x3f3f3f3f,mod=1e6;
struct Splay{
int rt=0,tot=0;
struct node{int ch[2],val,fa,sz,cnt;}t[N];
inline void push_up(int p){
t[p].sz=t[t[p].ch[0]].sz+t[t[p].ch[1]].sz+t[p].cnt;
}
inline void rotate(int x){
int y=t[x].fa,z=t[y].fa,k=(t[y].ch[1]==x);
t[z].ch[t[z].ch[1]==y]=x,t[x].fa=z;
t[y].ch[k]=t[x].ch[k^1],t[t[x].ch[k^1]].fa=y;
t[x].ch[k^1]=y,t[y].fa=x;
push_up(x),push_up(y);
}
inline void splay(int x,int goal){
while(t[x].fa!=goal){
int y=t[x].fa,z=t[y].fa;
if(z!=goal) (t[y].ch[0]==x)^(t[z].ch[0]==y)?rotate(x):rotate(y);
rotate(x);
}
if(goal==0) rt=x;
}
inline void insert(int x){
int u=rt,fa=0;
while(u&&t[u].val!=x) fa=u,u=t[u].ch[x>t[u].val];
if(u){t[u].cnt++; splay(u,0); return ;}
u=++tot; if(fa) t[fa].ch[x>t[fa].val]=u;
t[u].ch[0]=t[u].ch[1]=0,t[u].fa=fa,t[u].val=x,t[u].cnt=1,t[u].sz=1;
splay(u,0);
}
inline void find(int x){
int u=rt; if(!u) return ;
while(t[u].ch[x>t[u].val]&&x!=t[u].val) u=t[u].ch[x>t[u].val];
splay(u,0);
}
inline int Next(int x,int f){//前驱 0 ,后继 1
find(x); int u=rt;
if((t[u].val>=x&&f)||(t[u].val<=x&&!f)) return u;//(不严格后继)
u=t[u].ch[f]; while(t[u].ch[f^1]) u=t[u].ch[f^1];
return u;
}
inline int Next_une(int x,int f){
find(x); int u=rt;
if((t[u].val>x&&f)||(t[u].val<x&&!f)) return u;//(严格后继)
u=t[u].ch[f]; while(t[u].ch[f^1]) u=t[u].ch[f^1];
return u;
}
inline void del(int x){
int last=Next_une(x,0),next=Next_une(x,1);
splay(last,0),splay(next,last);
int son=t[next].ch[0];
if(t[son].cnt>1) t[son].cnt--,splay(son,0);
else t[next].ch[0]=0;
}
inline int kth(int x){
int u=rt; if(t[u].sz<x) return 0;
while(1){
int y=t[u].ch[0];
if(x>t[y].sz+t[u].cnt) x-=t[y].sz+t[u].cnt,u=t[u].ch[1];
else{if(t[y].sz>=x) u=y; else return t[u].val;}
}
}
inline int rank(int x){find(x); return t[t[rt].ch[0]].sz;}
}sp;
int res,n,cnt;
signed main(){
cin>>n; sp.insert(-inf),sp.insert(inf);
while(n--){
int op,x; scanf("%d %d",&op,&x);
if(op){
if(cnt>=0) sp.insert(x);
else{
int last=sp.t[sp.Next(x,0)].val,next=sp.t[sp.Next(x,1)].val;
if(x-last<=next-x) sp.del(last),res=(res+x-last)%mod;
else sp.del(next),res=(res+next-x)%mod;
}
cnt++;
}else{
if(cnt<=0) sp.insert(x);
else{
int last=sp.t[sp.Next(x,0)].val,next=sp.t[sp.Next(x,1)].val;
if(x-last<=next-x) sp.del(last),res=(res+x-last)%mod;
else sp.del(next),res=(res+next-x)%mod;
}
cnt--;
}
}
cout<<res<<endl;
return 0;
}
来源:CSDN
作者:青烟绕指柔!
链接:https://blog.csdn.net/weixin_43826249/article/details/103462965