FHQTreap刷题记录

  • 分裂：如图，按值分裂，u比k大则切黄色的一🔪，此时u接上Y的左子树，然后可见Y的左子树仍然空着，u也递归到其左子树。同理，u比k大则切红色的一🔪，此时u接上X的右子树，然后可见X的右子树仍然空着，u也递归到其右子树。
  • 合并：随机切红色和紫色，合并的是切口上面部分，return合并后的根，，比如，如果切红色，那递归合并X的右子树和Y，并且X的右儿子的指针指向X的右子树和Y合并的结果。

 

BZOJ3224: Tyvj 1728 普通平衡树

  • 模板题，代码： 
     1 #include <bits/stdc++.h>
 2 #define nmax 100010
 3  
 4 using namespace std;
 5 int s[nmax][2]={0};
 6 int si[nmax]={0}, v[nmax], r[nmax];
 7 int cnt = 0;
 8  
 9 void up(int p){
10     si[p] = si[ s[p][0] ] + si[ s[p][1] ] + 1;
11 }
12  
13 void spilit(int& x, int& y, int k, int u){
14     if(!u) { x=y=0; return; }
15     if(v[u] <= k){
16         x = u;
17         spilit(s[x][1], y, k, s[x][1]);
18     }else{
19         y = u;
20         spilit(x, s[y][0], k, s[y][0]);
21     }
22     up(u);
23 }
24  
25 int merge(int x, int y){
26     if(!x || !y){ return x|y; }
27     if(r[x] > r[y]) {
28         s[x][1] = merge(s[x][1], y);
29         up(x);
30         return x;
31     }else{
32         s[y][0] = merge(x, s[y][0]);
33         up(y);
34         return y;
35     }
36 }
37  
38 int kth(int k, int u){
39     int ans, t;
40     while(1){
41         t = si[ s[u][0] ] + 1;
42         if(k == t) { ans = u; break; }
43         else if(k < t) u=s[u][0];
44         else { u=s[u][1]; k-=t; }
45     }
46     return v[ans];
47 }
48  
49 int newnode(int k){
50     cnt++;
51     v[cnt] = k;
52     si[cnt] = 1;
53     r[cnt] = rand();
54     return cnt;
55 }
56  
57 int main(){
58     int n, op, num;
59     cin >> n;
60     int x, y, z, root=0;
61     while(n--){
62         scanf("%d%d", &op, &num);
63         switch (op)
64         {
65             case 1:
66                 z = newnode(num);
67                 spilit(x, y, num, root);
68                 root = merge( merge(x,z) , y );
69                 break;
70             case 2:
71                 spilit(x, y, num, root);
72                 spilit(x, z, num-1, x);
73                 z = merge(s[z][0], s[z][1]);
74                 root = merge( merge(x,z), y );
75                 break;
76             case 3:
77                 spilit(x, y, num-1, root);
78                 printf("%d\n", si[x]+1);
79                 root = merge(x, y);
80                 break;
81             case 4:
82                 printf("%d\n", kth(num, root) );
83                 break;
84             case 5:
85                 spilit(x, y, num-1, root);
86                 printf("%d\n", kth(si[x], x) );
87                 root = merge(x, y);
88                 break;
89             default:
90                 spilit(x, y, num, root);
91                 printf("%d\n", kth(1, y) );
92                 root = merge(x, y);
93                 break;
94         }
95     }
96     return 0;
97 }
    ( ఠൠఠ )ﾉ

     

posted @ 2019-11-20 00:43  连昵称都不能重复  阅读(186)  评论(0)    收藏  举报