[HNOI2004]宠物收养场（Treap）

洛谷传送门

这题真是恶心，一开始没理解题意。

原来如果有狗，狗就会存在收养场中，直到有人来领养；

　　如果有人，人也会存在收养场中，直到有狗来被领养。

就是建一个treap，狗来把狗插进去，人来后把狗领养完，多余的人就存在treap里，等狗来。

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

 using namespace std;

 int n, root, tot, pre, suc, flag = -, ans;
 int son[][], size[], rnd[], w[];

 inline void turn(int &k, int x)
 {
     int t = son[k][x];
     son[k][x] = son[t][x ^ ];
     son[t][x ^ ] = k;
     size[t] = size[k];
     size[k] = size[son[k][]] + size[son[k][]] + ;
     k = t;
 }

 inline void insert(int &k, int x)
 {
     if(!k)
     {
         k = ++tot;
         w[k] = x;
         rnd[k] = rand();
         size[k] = ;
         return;
     }
     size[k]++;
     if(w[k] < x)
     {
         insert(son[k][], x);
         if(rnd[k] > rnd[son[k][]]) turn(k, );
     }
     else
     {
         insert(son[k][], x);
         if(rnd[k] > rnd[son[k][]]) turn(k, );
     }
 }

 inline void get_pre(int k, int x)
 {
     if(!k) return;
     if(w[k] > x) get_pre(son[k][], x);
     else pre = k, get_pre(son[k][], x);
 }

 inline void get_suc(int k, int x)
 {
     if(!k) return;
     if(w[k] < x) get_suc(son[k][], x);
     else suc = k, get_suc(son[k][], x);
 }

 inline void del(int &k, int x)
 {
     if(!k) return;
     if(w[k] == x)
     {
         if(son[k][] * son[k][] == ) k = son[k][] + son[k][];
         else if(rnd[son[k][]] < rnd[son[k][]]) turn(k, ), del(k, x);
         else turn(k, ), del(k, x);
     }
     else
     {
         size[k]--;
         if(w[k] > x) del(son[k][], x);
         else del(son[k][], x);
     }
 }

 int main()
 {
     int i, a, b, x;
     scanf("%d", &n);
     for(i = ; i <= n; i++)
     {
         scanf("%d %d", &a, &b);
         if(flag == a) insert(root, b);
         else
         {
             if(size[root])
             {
                 pre = , get_pre(root, b);
                 suc = , get_suc(root, b);
                 if(!pre || !suc) x = pre + suc;
                 else if(b - w[pre] <= w[suc] - b) x = pre;
                 else x = suc;
                 ans = (ans + abs(w[x] - b)) % ;
                 del(root, w[x]);
             }
             else flag = a, insert(root, b);
         }
     }
     printf("%d", ans);
     return ;
 }