bzoj3224: Tyvj 1728 普通平衡树(spaly)

Posted 2020-10-30 CHerish_OI

3224: Tyvj 1728 普通平衡树

题目：传送门 

 

---

 

 

题解：

　　 啦啦啦啦又来敲个模版水经验啦~

 　　 

---

 

 

代码：

  1 #include<cstdio>
  2 #include<cstring>
  3 #include<cstdlib>
  4 #include<cmath>
  5 #include<algorithm>
  6 using namespace std;
  7 struct node
  8 {
  9     int d,c,n,f,son[2];
 10 }tr[110000];int len,root;
 11 void add(int d,int f)
 12 {
 13     len++;tr[len].d=d;tr[len].n=tr[len].c=1;
 14     tr[len].son[0]=tr[len].son[1]=0;tr[len].f=f;
 15     if(d<tr[f].d)tr[f].son[0]=len;
 16     else         tr[f].son[1]=len;
 17 }
 18 void update(int x)
 19 {
 20     int lc=tr[x].son[0],rc=tr[x].son[1];
 21     tr[x].c=tr[lc].c+tr[rc].c+tr[x].n;
 22 }
 23 int findip(int d)
 24 {
 25     int x=root;
 26     while(tr[x].d!=d)
 27     {
 28         if(d<tr[x].d)
 29         {
 30             if(tr[x].son[0]==0)break;
 31             else x=tr[x].son[0];
 32         }
 33         else
 34         {
 35             if(tr[x].son[1]==0)break;
 36             else x=tr[x].son[1];
 37         }
 38     }
 39     return x;
 40 }
 41 void rotate(int x,int w)
 42 {
 43     int f=tr[x].f,ff=tr[f].f;
 44     int r,R;
 45     r=tr[x].son[w],R=f;
 46     tr[R].son[1-w]=r;
 47     if(r!=0)tr[r].f=R;
 48      
 49     r=x,R=ff;
 50     if(tr[R].son[0]==f)tr[R].son[0]=r;
 51     else               tr[R].son[1]=r;
 52     tr[r].f=R;
 53      
 54     r=f,R=x;
 55     tr[R].son[w]=r;
 56     tr[r].f=R;
 57      
 58     update(f);update(x);
 59 }
 60 void splay(int x,int rt)
 61 {
 62     while(tr[x].f!=rt)
 63     {
 64         int f=tr[x].f,ff=tr[f].f;
 65         if(ff==rt)
 66         {
 67             if(tr[f].son[0]==x)rotate(x,1);
 68             else               rotate(x,0);
 69         }
 70         else
 71         {
 72                  if(tr[ff].son[0]==f && tr[f].son[0]==x)rotate(f,1),rotate(x,1);
 73             else if(tr[ff].son[1]==f && tr[f].son[1]==x)rotate(f,0),rotate(x,0);
 74             else if(tr[ff].son[1]==f && tr[f].son[0]==x)rotate(x,1),rotate(x,0);
 75             else if(tr[ff].son[0]==f && tr[f].son[1]==x)rotate(x,0),rotate(x,1);
 76         }
 77     }
 78     if(rt==0)root=x;
 79 }
 80 void ins(int d)
 81 {
 82     if(root==0)
 83     {
 84         add(d,0);root=len;
 85         return ;
 86     }
 87     int x=findip(d);
 88     if(tr[x].d==d)
 89     {
 90         tr[x].n++;
 91         update(x);
 92         splay(x,0);
 93     }
 94     else
 95     {
 96         add(d,x);
 97         update(x);
 98         splay(len,0);
 99     }
100 }
101 void del(int d)
102 {
103     int x=findip(d);if(tr[x].d!=d)return ;
104     splay(x,0);
105     if(tr[x].n>1){tr[x].n--;update(x);return ;}
106     if(tr[x].son[0]==0 && tr[x].son[1]==0){root=0;len=0;}
107     else if(tr[x].son[0]!=0 && tr[x].son[1]==0){root=tr[x].son[0];tr[root].f=0;}
108     else if(tr[x].son[0]==0 && tr[x].son[1]!=0){root=tr[x].son[1];tr[root].f=0;}
109     else
110     {
111         int p=tr[x].son[0];
112         while(tr[p].son[1]!=0)p=tr[p].son[1];
113         splay(p,x);
114          
115         int r=tr[x].son[1],R=p;
116         tr[R].son[1]=r;
117         tr[r].f=R;
118          
119         root=p;tr[root].f=0;
120         update(root);
121     }
122 }
123 void findpaiming(int d)
124 {
125     int x=findip(d);splay(x,0);int lc=tr[x].son[0];
126     printf("%d\n",tr[lc].c+1);
127 }
128 void findshuzi(int k)
129 {
130     if(tr[root].c<k){printf("-1\n");return ;}
131     int x=root;
132     while(1)
133     {
134         int lc=tr[x].son[0],rc=tr[x].son[1];
135         if(k<=tr[lc].c)x=lc;
136         else if(k>tr[lc].c+tr[x].n)k-=tr[lc].c+tr[x].n,x=rc;
137         else break;
138     }
139     printf("%d\n",tr[x].d);
140 }
141 void findqianqu(int d)
142 {
143     int x=findip(d);splay(x,0);
144     if(d<=tr[x].d && tr[x].son[0]!=0)
145     {
146         x=tr[x].son[0];
147         while(tr[x].son[1]!=0)x=tr[x].son[1];
148     }
149     if(d<=tr[x].d)x=0;
150     printf("%d\n",tr[x].d);
151 }
152 void findhouji(int d)
153 {
154     int x=findip(d);splay(x,0);
155     if(d>=tr[x].d && tr[x].son[1]!=0)
156     {
157         x=tr[x].son[1];
158         while(tr[x].son[0]!=0)x=tr[x].son[0];
159     }
160     if(d>=tr[x].d)x=0;
161     printf("%d\n",tr[x].d);
162 }
163 /*
164 void dell(int l,int r)
165 {
166     int lc=findqianqu(tr[l].d),rc=findhouji(tr[r].d);
167     spaly(lc,0);spaly(rc,lc);
168     tr[rc].son[0]=0;len-=tr[tr[rc].lc].c;
169     update(rc);update(lc);
170 }
171 */
172 int main()
173 {
174     int n;scanf("%d",&n);root=0;len=0;
175     for(int i=1;i<=n;i++)
176     {
177         int opt,d;scanf("%d%d",&opt,&d);
178         if(opt==1)ins(d);
179         else if(opt==2)del(d);
180         else if(opt==3)findpaiming(d);
181         else if(opt==4)findshuzi(d);
182         else if(opt==5)findqianqu(d);
183         else if(opt==6)findhouji(d);
184     }
185     return 0;
186 }