BZOJ_1208_&_Codevs_1258_[HNOI2004]_宠物收养所_(平衡树/set)

时间:2022-02-25 14:44:33

描述


http://www.lydsy.com/JudgeOnline/problem.php?id=1208

(据说codevs要更新?就不放codevs的地址了吧...)

有宠物和人,每个单位都有一个与其他单位不同的值(正),如果现在有宠物,来了一个人,人取走和他的值最接近的宠物(如果有两个选值小的),如果现在有人,来了一个宠物,和它值最接近的人(如果有两个选值小的)取走它.求每次带宠物走两个单位的值的差的绝对值的总和.

1208: [HNOI2004]宠物收养所

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 6798  Solved: 2677
[Submit][Status][Discuss]

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

分析


用平衡树查找前驱和后驱即可.依照题意,要么全是人要么全是宠物,所以建一棵树即可.模板题.

注意:

1.判断是否有前驱或后驱时要小心.

p.s.第一道用Splay写的题,调了好久...指针理解的不是很好啊.

Treap:

 #include <cstdio>
#include <cstdlib>
#include <algorithm>
using namespace std; const int oo=~0u>>; struct Treap{
struct node{
node* ch[];
int v,r,s;
node(int v,node* t):v(v){ ch[]=ch[]=t; r=rand(); s=; }
void push_up(){ s=ch[]->s+ch[]->s+; }
}*root,*null;
Treap(){
null=new node(,NULL); null->s=; null->r=oo;
root=null;
}
void rot(node* &o,bool d){
node* k=o->ch[!d]; o->ch[!d]=k->ch[d]; k->ch[d]=o;
o->push_up(); k->push_up(); o=k;
}
void insert(node* &o,int x){
if(o==null) o=new node(x,null);
else{
bool d=x>o->v;
insert(o->ch[d],x);
if(o->ch[d]->r<o->r) rot(o,!d);
o->push_up();
}
}
void remove(node* &o,int x){
if(o->v==x){
if(o->ch[]!=null&&o->ch[]!=null){
bool d=o->ch[]->r<o->ch[]->r;
rot(o,d); remove(o->ch[d],x);
o->push_up();
}
else{
node* u=o;
if(o->ch[]==null) o=o->ch[]; else o=o->ch[];
delete u;
}
}
else{
bool d=x>o->v;
remove(o->ch[d],x);
o->push_up();
}
}
int pre(int x){
node* t=root;
int ret=-;
while(t!=null){
if(t->v<x) ret=t->v,t=t->ch[];
else t=t->ch[];
}
return ret;
}
int suc(int x){
node* t=root;
int ret=-;
while(t!=null){
if(t->v>x) ret=t->v,t=t->ch[];
else t=t->ch[];
}
return ret;
}
}tree; const int maxn=+,mod=1e6;
int n;
int main(){
scanf("%d",&n);
int cnt=,now,ans=;
while(n--){
int a,b;
scanf("%d%d",&a,&b);
if(cnt==) now=a;
if(a==now){
tree.insert(tree.root,b);
cnt++;
}
else{
int pre=tree.pre(b);
int suc=tree.suc(b);
if(pre==-){
int t=suc-b;
ans=(ans+t)%mod;
tree.remove(tree.root,suc);
}
else if(suc==-){
int t=b-pre;
ans=(ans+t)%mod;
tree.remove(tree.root,pre);
}
else{
int t1=b-pre,t2=suc-b;
if(t1<=t2){
ans=(ans+t1)%mod;
tree.remove(tree.root,pre);
}
else{
ans=(ans+t2)%mod;
tree.remove(tree.root,suc);
}
}
cnt--;
}
}
printf("%d\n",ans);
return ;
} Treap

set:

 #include <bits/stdc++.h>
using namespace std; const int mod=;
int n,ans,a,b,now,cnt;
set <int> st;
int main(){
scanf("%d",&n);
set <int> :: iterator it;
while(n--){
scanf("%d%d",&a,&b);
if(!cnt) now=a;
if(a==now) st.insert(b), cnt++;
else{
it=st.lower_bound(b);
if(it!=st.end()&&it!=st.begin()){
int suc=*it,pre=*(--it);
if(b-pre<=suc-b){
st.erase(pre);
ans=(ans+b-pre)%mod;
}
else{
st.erase(suc);
ans=(ans+suc-b)%mod;
}
}
else if(it!=st.end()){
int suc=*it;
st.erase(suc);
ans=(ans+suc-b)%mod;
}
else{
int pre=*(--it);
st.erase(pre);
ans=(ans+b-pre)%mod;
}
cnt--;
}
}
printf("%d\n",ans);
return ;
}