BZOJ 1208 （可持久化Treap,合并与分裂操作）

1208: [HNOI2004]宠物收养所

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 4965   Solved: 1902
[ 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，最后一个领养者没有宠物可以领养)

/*

Treap的简单应用 不过需要注意几个细节！ 寻找前驱和后继节点的时候，是找的value值为v的严格前驱和非严格后继节点；

*/

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;
#define inf 0x0f0f0f0f
#define mod 1000000
int abs(int x){ return (x>=0?x:-x); }
struct Node {
    Node *ch[2];
    int rank,value,siz;
    Node(){ ch[0] = ch[1] = NULL; }
    Node(int v,int r) {
         ch[0] = ch[1] = NULL;
         value = v; rank = r;
         siz = 1;
    }
    void push_up() {
        siz = 1;
        if(ch[0]!=NULL) siz += ch[0]->siz;
        if(ch[1]!=NULL) siz += ch[1]->siz;
    }
}*root;
int res,tag;
typedef pair<Node *,Node *> DNode;
Node *merge(Node *(&node1),Node *(&node2)) {
    if(node1==NULL) return node2;
    if(node2==NULL) return node1;
    if(node1->rank > node2->rank){
        node1->ch[1] = merge(node1->ch[1],node2);
        node1->push_up();
        return node1;
    }
    else{
        node2->ch[0] = merge(node1,node2->ch[0]);
        node2->push_up();
        return node2;
    }
}
DNode split(Node *(&node),int k) {
    if(node==NULL) return DNode(NULL,NULL);
    if(k<=0)         return DNode(NULL,node);
    if(node->siz<=k) return DNode(node,NULL);
    DNode ans = DNode(NULL,NULL);
    if(node->ch[0]!=NULL&&node->ch[0]->siz >= k) {
        ans = split(node->ch[0],k);
        node->ch[0] = ans.second;
        node->push_up();
        ans.second = node;
    }
    else {
        int size = (node->ch[0]!=NULL?node->ch[0]->siz+1:1);
        ans = split(node->ch[1],k-size);
        node->ch[1] = ans.first;
        node->push_up();
        ans.first = node;
    }
    return ans;
}
//找node的严格前驱节点
Node *getPre(Node *(&node),int v) {
    if(node == NULL) return NULL;
    Node *ans = node;
    if(node->value >= v) ans = getPre(node->ch[0],v);
    else {
        ans = getPre(node->ch[1],v);
        if(ans==NULL) ans = node;
    }
    return ans;
}
//找node的非严格后继节点
Node *getNext(Node *(&node),int v) {
    if(node == NULL) return NULL;
    Node *ans = node;
    if(node->value >= v){
         ans = getNext(node->ch[0],v);
         if(ans==NULL) ans = node;
    }
    else ans = getNext(node->ch[1],v);
    return ans;
}
//查找严格<v节点个数
int search(Node *(&node),int v) {
    if(node == NULL) return 0;
    if(node->value >= v) return search(node->ch[0],v);
    else {
        int k = 1;
        if(node->ch[0]!=NULL) k += node->ch[0]->siz;
        return k + search(node->ch[1],v);
    }
}
void insert(int v) {
    int k = search(root,v);
    DNode tem = split(root,k);
    Node *node = new Node(v,rand());
    tem.first = merge(tem.first,node);
    root = merge(tem.first,tem.second);
}
void earse(int v){
    Node *left = getPre(root,v);
    Node *right = getNext(root,v);
    int k = search(root,v),ldis = inf,rdis = inf;
    if(left!=NULL)  ldis = v - left->value;
    if(right!=NULL) rdis = right->value - v;
    //printf("%d,%d\n",ldis,rdis);
    if(ldis<=rdis){
        res = (res + ldis%mod)%mod;
        DNode tem1 = split(root,k-1);
        DNode tem2 = split(tem1.second,1);
        delete tem2.first; tem2.first = NULL;
        root = merge(tem1.first,tem2.second);
    }
    else {
        res = (res + rdis%mod)%mod;
        DNode tem1 = split(root,k);
        DNode tem2 = split(tem1.second,1);
        delete tem2.first; tem2.first = NULL;
        root = merge(tem1.first,tem2.second);
    }
}
void solve(int flag,int v){
    if(root==NULL) tag = flag;
    if(flag==tag) insert(v);
    else          earse(v);
}
void delet(Node *(node)){
    if(node==NULL) return;
    delet(node->ch[0]);
    if(node->ch[0]!=NULL){
        delete node->ch[0];node->ch[0]=NULL;
    }
    delet(node->ch[1]);
    if(node->ch[1]!=NULL){
        delete node->ch[1];node->ch[1]=NULL;
    }
    delete node; node = NULL;
}
void init(){
    res = tag = 0;
    delet(root);
    root = NULL;
}
int main() {
    int n,flag,v;
    //freopen("Test.txt","r",stdin);
    while(~scanf("%d",&n)){
        init();
        for(int i = 1;i<=n;i++){
            scanf("%d%d",&flag,&v);
            solve(flag,v);
        }
        printf("%d\n",res);
    }
    return 0;
}