BZOJ 1208 [HNOI2004]宠物收养所

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)。【任务描述】你得到了一年当中，领养者和被收养宠物到来收养所的情况，希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。

---

【题目分析】
同理Treap复习第三题。
只需要用一个标记来记录当前的树中存放的是人还是宠物。

---

【代码】

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>

#include <set>
#include <queue>
#include <string>
#include <iostream>
#include <algorithm>

using namespace std;

int read()
{
    int x=0,f=1; char ch=getchar();
    while (ch<'0'||ch>'9') { if(ch=='-') f=-1; ch=getchar(); }
    while (ch>='0'&&ch<='9') { x=x*10+ch-'0'; ch=getchar(); }
    return x*f;
}

#define maxn (3000001)
#define ll long long
#define inf (0x3f3f3f3f)

int num[maxn],tim[maxn],val[maxn],lc[maxn],rc[maxn],tot=0,rt,tag=0;
int ansp,ansn,ans;

void lturn(int &k)
{
    int t=rc[k];
    rc[k]=lc[t];
    lc[t]=k;
    k=t;
    return ;
}

void rturn (int &k)
{
    int t=lc[k];
    lc[k]=rc[t];
    rc[t]=k;
    k=t;
    return ;
}

void ins(int &k,int x)
{
    if (!k)
    {
        k=++tot;
        tim[k]=1;
        num[k]=x;
        val[k]=rand();
        return ;
    }
    if (x==num[k])
    {
        tim[k]++;
        return ;
    }
    if (x<num[k])
    {
        ins(lc[k],x);
        if (val[lc[k]]>val[k]) rturn(k);
        return ;
    }
    if (x>num[k])
    {
        ins(rc[k],x);
        if (val[rc[k]]>val[k]) lturn(k);
        return ;
    }
}

void pre(int k,int x)
{
    if (!k) return ;
    if (num[k]==x) { ansp=x; return ; }
    if (num[k]<x) { ansp=num[k]; return pre(rc[k],x); }
    else return pre(lc[k],x);
}

void nxt(int k,int x)
{
    if (!k) return ;
    if (num[k]==x) { ansn=x; return ; }
    if (num[k]>x) { ansn=num[k]; return nxt(lc[k],x); }
    else return nxt(rc[k],x);
}

void del(int &k,int x)
{
    if (num[k]==x)
    {
        if (tim[k]>1)
        {
            tim[k]--;
            return ;
        }
        else
        {
            if (lc[k]*rc[k]==0) {k=lc[k]+rc[k]; return ;}
            if (val[lc[k]]>val[rc[k]]) {rturn(k); del(k,x); return ;}
            else {lturn(k); del(k,x); return;}
        }
    }
    if (num[k]>x) del(lc[k],x);
    else del(rc[k],x);
}

int main()
{
    srand(20000416);
    int m=read();
    ins(rt,inf); ins(rt,-inf);tag=0;
    while (m--)
    {
        int opt=read(),x=read();
        if (opt==tag) ins(rt,x);
        else
        {
            pre(rt,x); nxt(rt,x);
            if (ansp==-inf&&ansn==inf)
            {
                tag=opt;
                ins(rt,x);
                continue;
            }
            else
            {
                int tmp=abs(x-ansp);
                if (abs(ansn-x)<tmp)
                {
                    tmp=abs(ansn-x);
                    del(rt,ansn);
                    ans+=tmp;
                    ans%=1000000;
                    continue;
                }
                del(rt,ansp);
                ans+=tmp;
                ans%=1000000;
                continue;
            }
        }
        ans%=1000000;
    }
    printf("%d\n",ans);
}