hust 1007 treap

1. #include <iostream>
2. #include <algorithm>
3. using namespace std;
5. struct treap
6. {
7.     int key, pri, cnt;
8.     treap *lc, *rc;
9.     treap(int x, int y);
10. }NIL=treap(0,0), *nil=&NIL;
12. treap::treap(int x=0, int y=0)
13. {
14.     key=x,pri=y,cnt=1;
15.     lc=rc=nil;
16. }
18. typedef treap*node;
19. node root;
21. void right_rotate(node &y)
22. {
23.     node x = y->lc;
24.     y->lc = x->rc;
25.     x->rc = y;
26.     y = x;
27. }
29. void left_rotate(node &x)
30. {
31.     node y = x->rc;
32.     x->rc = y->lc;
33.     y->lc = x;
34.     x = y;
35. }
37. void insert(node &t, int x)
38. {
39.     if(t == nil)
40.     {
41.         t = new treap(x,rand()*rand());
42.     }
43.     else if(x == t->key) t->cnt++;
44.     else if(x < t->key)
45.     {
46.         insert(t->lc, x);
47.         if(t->lc->pri < t->pri) right_rotate(t);
48.     }
49.     else if(x > t->key)
50.     {
51.         insert(t->rc, x);
52.         if(t->rc->pri < t->pri) left_rotate(t);
53.     }
54. }
56. void del(node &t, int x)
57. {
58.     if(t == nil)return;
59.     if(x < t->key) del(t->lc, x);
60.     else if(x > t->key) del(t->rc, x);
61.     else
62.     {
63.         if(t->cnt > 1) {t->cnt--;return;}
64.         else if(t->lc==nil) t = t->rc;
65.         else if(t->rc==nil) t = t->lc;
66.         else
67.         {
68.             if(t->lc->pri < t->rc->pri)
69.             {
70.                 right_rotate(t);
71.                 del(t->rc,x);
72.             }
73.             else
74.             {
75.                 left_rotate(t);
76.                 del(t->lc,x);
77.             }
78.         }
79.     }
80. }
82. int succ(node t, int x)
83. {
84.     if(t==nil)return -1;
85.     if(x > t->key) return succ(t->rc, x);
86.     else if(x == t->key) return x;
87.     else
88.     {
89.         int tmp=succ(t->lc, x);
90.         if(tmp==-1)return t->key;
91.         else return tmp;
92.     }
93. }
95. int main()
96. {
97.     int i, j, k, t;
98.     int n, q, y;
100.     scanf("%d", &t);
101.     for(k = 1; k <= t; k++)
102.     {
103.         scanf("%d%d%d", &n, &q, &y);
104.         root = nil;
105.         for(i = 0; i < n; i++)
106.         {
107.             scanf("%d", &j);
108.             insert(root, j);
109.         }
110.         printf("Case %d:/n", k);
111.         for(i = 0; i < q; i++)
112.         {
113.             scanf("%d", &j);
114.             int tmp = succ(root, j);
115.             if(tmp == -1 || tmp-j>y) puts("-1");
116.             else
117.             {
118.                 del(root, tmp);
119.                 printf("%d/n", tmp);
120.             }
121.         }
122.     }
123. }