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);}