【bzoj】3224: Tyvj 1728 普通平衡树

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<algorithm>
  4 #include<vector>
  5 #include<cstdlib>
  6 #include<cmath>
  7 #include<cstring>
  8 using namespace std;
  9 #define maxn 110010
 10 #define llg  int
 11 #define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
 12 int n,m,ans;
 13 
 14 inline int getint()
 15 {
 16        int w=0,q=0;
 17        char c=getchar();
 18        while((c<‘0‘ || c>‘9‘) && c!=‘-‘) c=getchar();
 19        if (c==‘-‘)  q=1, c=getchar();
 20        while (c>=‘0‘ && c<=‘9‘) w=w*10+c-‘0‘, c=getchar();
 21        return q ? -w : w;
 22 }
 23 
 24 struct MYTREAP
 25 {
 26     int root,size;
 27     struct 
 28     {
 29         int l,r,rnd,val,w,size;
 30     }po[maxn];
 31     
 32     void init(){ memset(po,0,sizeof(po)); root=size=0;}
 33     
 34     void update(llg now) {po[now].size=po[po[now].l].size+po[po[now].r].size+po[now].w;}
 35         
 36     void right_(llg &now) {llg t=po[now].l; po[now].l=po[t].r; po[t].r=now; po[t].size=po[now].size; update(now); now=t;}
 37     void left_(llg &now)  {llg t=po[now].r; po[now].r=po[t].l; po[t].l=now; po[t].size=po[now].size; update(now); now=t;}
 38 
 39     void in(llg val)
 40     {
 41         insert(root,val);
 42     }
 43 
 44     void insert(llg &now,llg val)
 45     {
 46         if (now==0)
 47         {
 48             now=++size;
 49             po[now].size=po[now].w=1; po[now].val=val; po[now].rnd=rand();
 50             return ;
 51         }
 52         po[now].size++;
 53         if (po[now].val==val)
 54         {
 55             po[now].w++;
 56         }
 57         else
 58             if (po[now].val<val)
 59             {
 60                 insert(po[now].r,val);
 61                 if (po[po[now].r].rnd<po[now].rnd) left_(now);
 62             }
 63             else
 64             {
 65                 insert(po[now].l,val);
 66                 if (po[po[now].l].rnd<po[now].rnd) right_(now);
 67             }
 68     }
 69     
 70     void del(llg &now,llg val)
 71     {
 72         if (now==0) return ;
 73         if (po[now].val==val)
 74         {
 75             if (po[now].w>1)
 76             {
 77                 po[now].w--; po[now].size--;
 78                 return ;
 79             }
 80             if (po[now].l*po[now].r==0) now=po[now].l+po[now].r;
 81             else
 82             {
 83                 if (po[po[now].l].rnd<po[po[now].r].rnd) right_(now); else left_(now);
 84                 del(now,val);
 85             }
 86         }
 87         else
 88         {
 89             po[now].size--;
 90             if (val>po[now].val) del(po[now].r,val);
 91             else del(po[now].l,val);
 92         }
 93     }
 94 
 95     int ask_rank(llg now,llg val)
 96     {
 97         if (now==0) return 0;
 98         if (po[now].val==val) return po[po[now].l].size+1;
 99         else
100             if (val>po[now].val) return po[po[now].l].size+po[now].w+ask_rank(po[now].r,val);
101             else return ask_rank(po[now].l,val);
102     }
103 
104     int ask_num(llg now,int val)
105     {
106         if (now==0) return 0;
107         if (val<=po[po[now].l].size) 
108             return ask_num(po[now].l,val);
109         else if (val>po[po[now].l].size+po[now].w) 
110             return ask_num(po[now].r,val-po[po[now].l].size-po[now].w);
111         else 
112             return po[now].val; 
113     }
114 
115     void pre(llg now,llg val)
116     {
117         if (now==0) return ;
118         if (po[now].val<val)
119         {
120             ans=now; 
121             pre(po[now].r,val);
122         }
123         else pre(po[now].l,val);
124     }
125 
126     void nex(llg now,llg val)
127     {
128         if (now==0) return ;
129         if (po[now].val>val)
130         {
131             ans=now;
132             nex(po[now].l,val);
133         }
134         else nex(po[now].r,val);
135     }
136 
137 }tree;
138 
139 int main()
140 {
141     yyj("3224");
142     srand(1346);
143     tree.init();
144     cin>>n;
145     llg ty,x;
146     for (llg i=1;i<=n;i++)
147     {
148         ty=getint(),x=getint();
149         switch(ty)
150         {
151             case 1:tree.in(x);break;
152             case 2:tree.del(tree.root,x);break;
153             case 3:printf("%d\n",tree.ask_rank(tree.root,x));break;
154             case 4:printf("%d\n",tree.ask_num(tree.root,x));break;
155             case 5:ans=0;tree.pre(tree.root,x);printf("%d\n",tree.po[ans].val);break;
156             case 6:ans=0;tree.nex(tree.root,x);printf("%d\n",tree.po[ans].val);break;
157         }
158     }
159     return 0;
160 }