HNOI2004 宠物收养所 解题报告

　　首先读完这题第一印象，是个裸题，很高兴。其次在打完代码之后，第二印象，很恶心，Treap的代码太长了，我今天下午敲了三遍，手都麻了。

　　废话不多说，正题。其实这个题不难，有几个点是很好的，首先，他的a值没有重复的，这就保证了你找前驱和后继的正确性，再次，没有宠物和人会同时在收养所内，那么你找的前驱和后继就不用判断是人还是动物。再次，就是AC了。

　　这个题一开始全WA，是因为没有注意前驱和后继不能为0的情况，导致自己领养自己或者自己被自己领养的情况出现，改了之后AC。

　　Treap的裸题，建议码一下。

　　直接上代码。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdlib>

using namespace std;
const int maxn = 100000 + 5;

int N,flag,temp;
int Ans,size,root;
int Out = 0;
int ani_cnt = 0,er_cnt = 0;
int pre,bac;

struct data{
    int l,r,w,v,size,rnd;
}tr[maxn];
void update(int k){
    tr[k].size = tr[tr[k].l].size + tr[tr[k].r].size + tr[k].w;
}
void rturn(int &k){
    int t = tr[k].l;tr[k].l = tr[t].r;tr[t].r = k;
    tr[t].size = tr[k].size;update(k);k = t;
}
void lturn(int &k){
    int t = tr[k].r;tr[k].r = tr[t].l;tr[t].l = k;
    tr[t].size = tr[k].size;update(k);k = t;
}
void insert(int &k,int x){
    if(k == 0){
        size ++;k = size;
        tr[k].w = tr[k].size = 1;
        tr[k].v = x;tr[k].rnd = rand();
        return;
    }
    tr[k].size ++;
    if(tr[k].v == x){
        tr[k].w ++;
    }
    else if(x < tr[k].v){
        insert(tr[k].l,x);
        if(tr[tr[k].l].rnd < tr[k].rnd)
            rturn(k);
    }
    else {
        insert(tr[k].r,x);
        if(tr[tr[k].r].rnd < tr[k].rnd)
            lturn(k);
    }
}
void del(int &k,int x){
    if(k == 0)
        return;
    if(tr[k].v == x){
        if(tr[k].w > 1){
            tr[k].w --;tr[k].size --;
            return;
        }
        if(tr[k].r*tr[k].l == 0)
            k = tr[k].r + tr[k].l;
        else if(tr[tr[k].l].rnd < tr[tr[k].r].rnd)
            rturn(k),del(k,x);
        else
            lturn(k),del(k,x);
    }
    else if(x > tr[k].v)
        tr[k].size --,del(tr[k].r,x);
    else
        tr[k].size --,del(tr[k].l,x);
}
int query_rank(int k,int x){
    if(k == 0)
        return 0;
    if(x == tr[k].v)
        return tr[tr[k].l].size + 1;
    else if(x > tr[k].v)
        return tr[tr[k].l].size + tr[k].w + query_rank(tr[k].r,x);
    else  return query_rank(tr[k].l,x);
}
int query_num(int k,int x){
    if(k == 0)
        return 0;
    if(x <= tr[tr[k].l].size)
        return query_num(tr[k].l,x);
    else if(x > tr[tr[k].l].size + tr[k].w)
        return query_num(tr[k].r,x-tr[tr[k].l].size-tr[k].w);
    else  return tr[k].v;
}
void query_pre(int k,int x){
    if(k == 0)
        return;
    if(x > tr[k].v){
        Ans = k;query_pre(tr[k].r,x);
    }
    else  query_pre(tr[k].l,x);
}
void query_sub(int k,int x){
    if(k == 0)
        return;
    if(x < tr[k].v){
        Ans = k;query_sub(tr[k].l,x);
    }
    else  query_sub(tr[k].r,x);
}

int main(){
    freopen("pet.in","r",stdin);
    freopen("pet.out","w",stdout);
    scanf("%d",&N);
    for(int i = 1;i <= N;++ i){
        scanf("%d%d",&flag,&temp);
        if(!flag){
            insert(root,temp);
            ani_cnt ++;
            if(er_cnt >= ani_cnt){//如果人多动物少，那么这个动物立刻被领养  
                query_pre(root,temp);pre = tr[Ans].v;Ans = 0;
                query_sub(root,temp);bac = tr[Ans].v;Ans = 0;
                if(pre == bac){
                    Out += abs(pre-temp);
                    del(root,pre);del(root,temp);    
                }
                else{
                    if(temp-pre < bac-temp){
                        if(pre){
                            Out += temp-pre;
                            del(root,pre);del(root,temp);
                        }
                        else{
                            Out += bac-temp;
                            del(root,bac);del(root,temp);
                        }
                        
                    }
                    else{
                        if(bac){
                            Out += bac-temp;
                            del(root,bac);del(root,temp);
                        }
                        else{
                            Out += temp-pre;
                            del(root,pre);del(root,temp);
                        }
                    }
                }
                --er_cnt;--ani_cnt;
            }
        }
        else{
            insert(root,temp);
            er_cnt ++;
            if(ani_cnt >= er_cnt){//如果动物多人少，那么这个人立刻领养一个动物 
                query_pre(root,temp);pre = tr[Ans].v;Ans = 0;
                query_sub(root,temp);bac = tr[Ans].v;Ans = 0;
                if(pre == bac){
                    Out += abs(pre-temp);
                    del(root,pre);del(root,temp);
                }
                else{
                    if(temp-pre < bac-temp){
                        if(pre){
                            Out += temp-pre;
                            del(root,pre);del(root,temp);
                        }
                        else{
                            Out += bac-temp;
                            del(root,bac);del(root,temp);
                        }
                    }
                    else{
                        if(bac){
                            Out += bac-temp;
                            del(root,bac);del(root,temp);
                        }
                        else{
                            Out += temp-pre;
                            del(root,pre);del(root,temp);
                        }
                    }
                }
                --er_cnt;--ani_cnt;
            }
        }
        Out %= 1000000;
    }    
    printf("%d",Out%1000000);
    fclose(stdin);fclose(stdout);
    return 0;
}