【模板】 普通平衡树

emm这两天终于来学学平衡树了，自己没有板，只能靠网上大佬的板来学了，感觉大佬打的板挺好的。

原文：https://blog.csdn.net/a_comme_amour/article/details/79382104

评测的是洛谷：https://www.luogu.org/record/22239881

  1 #include<bits/stdc++.h>
  2 using namespace std;
  3 const int N=1e5+8;
  4 int ch[N][2],f[N],siz[N],cnt[N],val[N];
  5 int sz,rt;
  6 inline void clear(int x){
  7     ch[x][0]=ch[x][1]=f[x]=siz[x]=cnt[x]=val[x]=0;
  8 }
  9 inline bool get(int x){
 10     return ch[f[x]][1]==x;
 11 }
 12 inline void update(int x){
 13     if(x){
 14         siz[x]=cnt[x];
 15         if(ch[x][0]) siz[x]+=siz[ch[x][0]];
 16         if(ch[x][1]) siz[x]+=siz[ch[x][1]];
 17     }
 18 }
 19 inline void rotate(int x){
 20     int fa=f[x],ffa=f[fa],which=get(x);
 21     ch[fa][which]=ch[x][which^1];f[ch[fa][which]]=fa;
 22     ch[x][which^1]=fa;f[fa]=x;
 23     f[x]=ffa;
 24     if(ffa) ch[ffa][ch[ffa][1]==fa]=x;
 25     update(fa);update(x);
 26 }
 27 inline void splay(int x){
 28     for(int fa;fa=f[x];rotate(x)){
 29         if(f[fa]){
 30             rotate( (get(x)==get(fa)) ? fa : x );
 31         }
 32     }
 33     rt=x;
 34 }
 35 inline void ins(int x){
 36     if(!rt){
 37         ++sz;ch[sz][0]=ch[sz][1]=f[sz]=0;
 38         rt=sz;siz[sz]=cnt[sz]=1;val[sz]=x;
 39         return;
 40     }
 41     int now=rt,fa=0;
 42     while(1){
 43         if(x==val[now]){
 44             ++cnt[now];
 45             update(now);update(fa);
 46             splay(now);
 47             return;
 48         }
 49         fa=now;
 50         now=ch[now][val[now]<x];
 51         if(!now){
 52             ++sz;
 53             ch[sz][0]=ch[sz][1]=0;
 54             f[sz]=fa;
 55             siz[sz]=cnt[sz]=1;
 56             ch[fa][val[fa]<x]=sz;
 57             val[sz]=x;
 58             update(fa);
 59             splay(sz);
 60             return;
 61         }
 62     }
 63 }
 64 inline int rnk(int x){
 65     int now=rt,ans=0;
 66     while(1){
 67         if(x<val[now]) now=ch[now][0];
 68         else{
 69             ans+=(ch[now][0]?siz[ch[now][0]]:0);
 70             if(x==val[now]){
 71                 splay(now);
 72                 return ans+1;
 73             }
 74             ans+=cnt[now];
 75             now=ch[now][1];
 76         }
 77     }
 78 }
 79 inline int kth(int x){
 80     int now=rt;
 81     while(1){
 82         if(ch[now][0] && x<=siz[ch[now][0]]) now=ch[now][0];
 83         else{
 84             int tem=(ch[now][0]?siz[ch[now][0]]:0)+cnt[now];
 85             if(x<=tem) return val[now];
 86             x-=tem;now=ch[now][1];
 87         }
 88     }
 89 }
 90 inline int pre(){
 91     int now=ch[rt][0];
 92     while(ch[now][1]) now=ch[now][1];
 93     return now;
 94 }
 95 inline int nex(){
 96     int now=ch[rt][1];
 97     while(ch[now][0]) now=ch[now][0];
 98     return now;
 99 }
100 inline void del(int x){
101     rnk(x);
102     if(cnt[rt]>1){--cnt[rt]; update(rt); return;}
103     if(!ch[rt][0] && !ch[rt][1]) {clear(rt); rt=0; return;}
104     if(!ch[rt][0]){
105         int oldrt=rt; rt=ch[rt][1]; f[rt]=0;
106         clear(oldrt); return;
107     }
108     if(!ch[rt][1]){
109         int oldrt=rt; rt=ch[rt][0]; f[rt]=0;
110         clear(oldrt); return;
111     }
112     int lebig=pre(),oldrt=rt;
113     splay(lebig);
114     ch[rt][1]=ch[oldrt][1];
115     f[ch[oldrt][1]]=rt;
116     clear(oldrt);
117     update(rt);
118 }
119 int main(){
120     int n;scanf("%d",&n);
121     for(int i=1;i<=n;++i){
122         int op,x;scanf("%d %d",&op,&x);
123         switch(op){
124             case 1:ins(x);break;
125             case 2:del(x);break;
126             case 3:printf("%d\n",rnk(x));break;
127             case 4:printf("%d\n",kth(x));break;
128             case 5:ins(x);printf("%d\n",val[pre()]);del(x);break;
129             case 6:ins(x);printf("%d\n",val[nex()]);del(x);break;
130         }
131     }
132     return 0;
133 }

View Code