BZOJ 1208

来源：互联网发布：手机淘宝首页尺寸大小编辑：程序博客网时间：2024/05/21 13:56

我表示只用一棵 Splay 。

因为题目里明确说明了“同一时间呆在收养所中的，要么全是宠物，要么全是领养者”。

每个节点存一个 kind 值。

找前驱找后继删节点删根，不说了。。。

删根的时候没判有没有儿子WA了几次。。。

丑陋的代码贴出。。。

#include <iostream>#include <cstdio>#include <cstring>const int INF=1<<29;const int M=1000000;const int N=80100;int ch[N][2],key[N],kind[N],pre[N];int n,i,a,b,tot,rt,ans;void newnode(int x,int k,int fa){     key[++tot]=x;     kind[tot]=k;     pre[tot]=fa;     ch[fa][x>=key[fa]]=tot;     ch[tot][0]=ch[tot][1]=0;}void rotate(int x,int k){     int y=pre[x];     ch[y][!k]=ch[x][k];     pre[ch[x][k]]=y;     if (pre[y]) ch[pre[y]][ch[pre[y]][1]==y]=x;     pre[x]=pre[y];     ch[x][k]=y;     pre[y]=x;}void splay(int x,int goal){     while (pre[x]!=goal)     {           int y=pre[x];           if (pre[y]==goal) rotate(x,ch[y][0]==x);           else           {               int k=ch[pre[y]][0]==y;               if (ch[y][k]==x)               {                  rotate(x,!k);                  rotate(x,k);               }               else               {                   rotate(y,k);                   rotate(x,k);               }           }     }     if (!goal) rt=x;}void ins(int k,int x){     if (!rt)     {        newnode(x,k,0);        rt=tot;        return;     }     int r=rt;     while (ch[r][x>=key[r]]) r=ch[r][x>=key[r]];     int f=kind[r]^k;     newnode(x,k,r);     splay(tot,0);     if (f)     {        int tl=ch[rt][0];        int tr=ch[rt][1];        if (tl) while (ch[tl][1]) tl=ch[tl][1];        if (tr) while (ch[tr][0]) tr=ch[tr][0];        int t1=tl?key[rt]-key[tl]:INF;        int t2=tr?key[tr]-key[rt]:INF;        if (t1>t2)        {           ans=(ans+t2)%M;           splay(tr,rt);           ch[rt][1]=ch[tr][1];           if (ch[tr][1]) pre[ch[tr][1]]=rt;           pre[tr]=key[tr]=ch[tr][0]=ch[tr][1]=0;           if (ch[rt][0]==0&&ch[rt][1]==0)           {              rt=0;              return;           }           int tmp=rt;           if (ch[tmp][0]==0)           {              rt=ch[tmp][1];              pre[rt]=pre[tmp]=key[tmp]=ch[tmp][0]=ch[tmp][1]=0;              return;           }           r=ch[tmp][1];           rt=ch[tmp][0];           pre[rt]=pre[tmp]=key[tmp]=ch[tmp][0]=ch[tmp][1]=0;           if (tl) ch[tl][1]=r;           if (r) pre[r]=tl;           if (tl) splay(tl,0);        }        else        {            ans=(ans+t1)%M;            splay(tl,rt);            ch[rt][0]=ch[tl][0];            if (ch[tl][0]) pre[ch[tl][0]]=rt;            key[tl]=ch[tl][0]=ch[tl][1]=0;            if (ch[rt][0]==0&&ch[rt][1]==0)            {               rt=0;               return;            }            int tmp=rt;            if (ch[tmp][1]==0)            {               rt=ch[tmp][0];               pre[rt]=pre[tmp]=key[tmp]=ch[tmp][0]=ch[tmp][1]=0;               return;            }            r=ch[tmp][0];            rt=ch[tmp][1];            pre[rt]=pre[tmp]=key[tmp]=ch[tmp][0]=ch[tmp][1]=0;            if (tr) ch[tr][0]=r;            if (r) pre[r]=tr;            if (tr) splay(tr,0);        }     }}int main(){    scanf("%d",&n);    for (i=1;i<=n;i++)    {        scanf("%d%d",&a,&b);        ins(a,b);    }    printf("%d\n",ans);    return 0;}

0 0