HYSBZ 1208 宠物收养所 （Splay树）

题意：一家宠物收养所负责处理领养者与遗弃宠物业务，有人来领宠物，则领一只最理想的。若没有宠物了，领养者们就得等到宠物来，宠物一来立刻送给其中一个等待者。如果有两个理想的选择，则选择那个值较小的。收养所时刻只存在人or宠物or没有存在任何！

思路：

　　这题主要是树中可能存在人或者宠物，但两者不可能同时存在，因为一旦存在，必须有对象被配对并从树中删除。 所有函数独自写起来都是很简单的，但是合起来使用时就得小心了，得配合一点。

　　要时刻注意树中是否还有节点，若没有，则可能要树要换种类了。

 #include <bits/stdc++.h>
 #define pii pair<int,int>
 #define INF 0x3f7f7f7f
 #define LL long long
 using namespace std;
 const int N=;
 const int mod=;

 struct node
 {
     int key, pre, ch[];
 }nod[N];
 int node_cnt, root, flag;

 int create_node(int v,int far)
 {
     nod[node_cnt].ch[]=nod[node_cnt].ch[]=;
     nod[node_cnt].pre=far;
     nod[node_cnt].key=v;
     return node_cnt++;
 }

 void Rotate(int t,int d)
 {
     int son=nod[t].ch[d];
     int far=nod[t].pre;
     int gra=nod[far].pre;
     nod[t].pre=gra;
     nod[far].pre=t;
     nod[son].pre=far;
     nod[t].ch[d]=far;
     nod[far].ch[d^]=son;
     nod[gra].ch[ nod[gra].ch[]==far ]=t;
 }

 int Insert(int t,int v)
 {
     if(t==)    return  root=create_node(v, ); //树中还没有节点。
     if( v < nod[t].key )
     {
         if(nod[t].ch[])  return Insert(nod[t].ch[], v);
         else              return nod[t].ch[]=create_node(v, t);
     }
     else
     {
         if(nod[t].ch[])  return Insert(nod[t].ch[], v);
         else              return nod[t].ch[]=create_node(v, t);
     }
 }

 void Splay(int t, int goal)
 {
     while(nod[t].pre!=goal)
     {
         int f=nod[t].pre, g=nod[f].pre;
         if(g==goal)    Rotate( t, nod[f].ch[]==t );
         else
         {
             int d1=nod[f].ch[]==t, d2=nod[g].ch[]==f;
             if(d1==d2)    Rotate(f, d1),Rotate(t, d1);
             else          Rotate(t, d1),Rotate(t, d2);
         }
     }
     if(goal==) root=t; //随时更新根！
 }

 int Find(int t,int v)   //在子树t中找到值为v的点，返回点号
 {
     while(t)
     {
         if(v==nod[t].key)   return t;
         if(v<nod[t].key)    t=nod[t].ch[];
         else                t=nod[t].ch[];
     }
     return ;   //找不到
 }

 int Find_pre(int t,int v)
 {
     int val=-;
     while(t)
     {
         if(nod[t].key<v)    val=max(val, nod[t].key);
         if(v<nod[t].key)    t=nod[t].ch[];
         else                t=nod[t].ch[];
     }
     if(val<)   return INF;
     return val;
 }

 int Find_bac(int t,int v)
 {
     int val=INF;
     while(t)
     {
         if(nod[t].key>v)    val=min(val, nod[t].key);
         if(v<nod[t].key)    t=nod[t].ch[];
         else                t=nod[t].ch[];
     }
     return val;
 }

 void Delete(int v)  //在树中删除值为v的任意一个点
 {
     Splay(Find(root, v), );   //将目标旋转到根
     int L=nod[root].ch[];
     int R=nod[root].ch[];
     if(L== && R==)    root=;
     else if(R==)    //没有右子树
     {
         nod[L].pre=;
         root=L;
     }
     else if(L==)    //没有左子树
     {
         nod[R].pre=;
         root=R;
     }
     else            //有前有后
     {
         while(nod[R].ch[]) R=nod[R].ch[]; //找到后继
         Splay(R, root);
         nod[R].ch[]=L;
         nod[R].pre=;
         nod[L].pre=R;
         root=R;
     }
 }

 int main()
 {
     //freopen("input.txt", "r", stdin);
     int n, ans, a, b, contain;
     while(cin>>n)
     {
         flag=node_cnt=;
         contain=root=ans=;
         while(n--)
         {
             scanf("%d%d",&a,&b);
             if(a==flag || contain==)    // 树为空 or 树中是同种类 则 插入
             {
                 flag=a;
                 contain++;
                 Splay(Insert(root, b), );
             }
             else
             {
                 if(Find(root, b))    Delete(b); //刚好有值为b的。
                 else
                 {
                     int L=Find_pre(root, b);
                     int R=Find_bac(root, b);
                     if( abs(L-b)<=abs(R-b) )      //L和R不可能同时为INF。
                     {
                         ans=(ans+abs(L-b))%mod;
                         Delete(L);
                     }
                     else
                     {
                         ans=(ans+abs(R-b))%mod;
                         Delete(R);
                     }
                 }
                 contain--;
             }
         }
         printf("%d\n", ans );
     }
     return ;
 }

AC代码