Bzoj1208 [HNOI2004]宠物收养所

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 7457  Solved: 2960

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

平衡树-Splay

在不同的时间，树可能有两种不同的形态：存宠物信息的树或者存客人信息的树。

对于一行新读入，如果读入物的属性和树属性相同，则将插入树中，否则从树中匹配一个最优位置，并将该位置从树中删除。

Splay维护树上信息，查找前驱后继即可。

花了一个小时艰难地敲对splay代码，提交以后一看，3000+ms，别人都是100+ms

常数写挂了，感到十分害怕。

花了近两个小时调试各种地方，感觉都没有问题，这时突然想到是不是splay转少了？

↑果不其然，在132行的位置 Splay(cnt)写成了Splay(root)，这是道splay裸题，数据肯定会卡长链……改完之后妥妥100ms跑完

↑喜闻乐见浪费青春2333333333

 /*by SilverN*/
 #include<algorithm>
 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #define LL long long
 using namespace std;
 const int INF=;
 const int mod=;
 const int mxn=;
 int read(){
     int x=,f=;char ch=getchar();
     while(ch<'' || ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>='' && ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 int n;
 LL ans;
 struct node{
     int ch[],fa;
     int v;
 }t[mxn];
 int cnt=,root=;
 void newnode(int &rt,int w,int fa){
     rt=++cnt;
     t[rt].ch[]=t[rt].ch[]=;
     t[rt].fa=fa;
     t[rt].v=w;
     return;
 }
 void rotate(int x,int kind){
     int y=t[x].fa;int z=t[y].fa;
     t[y].ch[^kind]=t[x].ch[kind];
     if(t[x].ch[kind]) t[t[x].ch[kind]].fa=y;
     t[x].fa=z;
     if(z)t[z].ch[t[z].ch[]==y]=x;
     t[y].fa=x;t[x].ch[kind]=y;
     return;
 }
 void Splay(int rt){
     while(t[rt].fa){
         int y=t[rt].fa; int z=t[y].fa;
         if(!z)rotate(rt,t[y].ch[]!=rt);
         else{
             if(t[z].ch[]==y){
                 if(t[y].ch[]==rt){rotate(y,);rotate(rt,);}
                 else{rotate(rt,);rotate(rt,);}
             }
             else{
                 if(t[y].ch[]==rt){rotate(y,);rotate(rt,);}
                 else{rotate(rt,);rotate(rt,);}
             }
         }
     }
     root=rt;
     return;
 }
 void insert(int v){//插入值
     if(!root){newnode(root,v,);return;}//建立新结点
     int now=root;
     while(){
         if(v>t[now].v){
             if(t[now].ch[]==){newnode(t[now].ch[],v,now);return;}
             else now=t[now].ch[];
         }
         else{
             if(t[now].ch[]==){newnode(t[now].ch[],v,now);return;}
             else now=t[now].ch[];
         }
     }
     return;
 }
 void Del(int x){
     Splay(x);
     //将x提为根结点
     if(!t[x].ch[] && !t[x].ch[]){root=;return;}
     //x没有子树，删除之后树变成空树
     if(!t[x].ch[]){root=t[x].ch[];t[root].fa=;return;}//x没有左子树，直接删除
     if(!t[x].ch[]){root=t[x].ch[];t[root].fa=;return;}//x没有右子树，直接删除
     int tmp=t[x].ch[];
     while(t[tmp].ch[])tmp=t[tmp].ch[];//找到x左子树的最右结点
     t[t[x].ch[]].fa=;
     Splay(tmp);
     t[tmp].ch[]=t[x].ch[];
     t[t[x].ch[]].fa=tmp;
     root=tmp;
     return;
 }
 int find_pre(int v){
     int now=root;
     int pos=,tmp=INF;
     while(){
         if(!now)return pos;
         if(t[now].v<v && (v-t[now].v<tmp) ){
             tmp=v-t[now].v;
             pos=now;
         }
         if(t[now].v<v)now=t[now].ch[];
         else now=t[now].ch[];
     }
     return pos;
 }
 int find_suc(int v){
     int now=root;
     int pos=,tmp=INF;
     while(){
         if(!now)return pos;
         if(t[now].v>v && (t[now].v-v<tmp) ){
             tmp=t[now].v-v;
             pos=now;
         }
         if(t[now].v<v)now=t[now].ch[];
         else now=t[now].ch[];
     }
     return pos;
 }
 int main(){
     n=read();
     int i,j,a,b;
     int type=;
     type=read();//初始类型
     j=read();//第一个值
     newnode(root,j,);
     ans=;
     for(i=;i<=n;i++){
         a=read();b=read();
         if(!root){
             type=a;//更改当前树类型
             newnode(root,b,);
             continue;
         }
         if(a==type){insert(b); Splay(cnt);}
         else{
             int pre=find_pre(b);
             int suc=find_suc(b);
             if(pre || suc){
                 int res1=INF; int res2=INF;
                 if(pre)res1=abs(b-t[pre].v);
                 if(suc)res2=abs(b-t[suc].v);
                 if(res1<=res2){ans=(ans+res1)%mod;Del(pre);}//比较前驱和后继
                 else{ans=(ans+res2)%mod;Del(suc);}
             }
         }
     }
     printf("%lld\n",ans);
     return ;
 }