3224: Tyvj 1728 普通平衡樹

Time Limit: 10 Sec Memory Limit: 128 MB

Description

您需要寫一種數(shù)據(jù)結(jié)構(gòu)（可參考題目標(biāo)題），來維護(hù)一些數(shù)，其中需要提供以下操作：
1. 插入x數(shù)
2. 刪除x數(shù)(若有多個(gè)相同的數(shù)，因只刪除一個(gè))
3. 查詢x數(shù)的排名(若有多個(gè)相同的數(shù)，因輸出最小的排名)
4. 查詢排名為x的數(shù)
5. 求x的前驅(qū)(前驅(qū)定義為小于x，且最大的數(shù))
6. 求x的后繼(后繼定義為大于x，且最小的數(shù))

Input

第一行為n，表示操作的個(gè)數(shù),下面n行每行有兩個(gè)數(shù)opt和x，opt表示操作的序號(1<=opt<=6)

Output

對于操作3,4,5,6每行輸出一個(gè)數(shù)，表示對應(yīng)答案

Sample Input

10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598

Sample Output

106465
84185
492737

HINT

1.n的數(shù)據(jù)范圍：n<=100000
2.每個(gè)數(shù)的數(shù)據(jù)范圍：[-2e9,2e9]

題解

裸的平衡樹，因?yàn)門reap好寫，所以就寫Treap

代碼如下

//BZOJ 3224 #include<cstdio> #include<cctype> #include<cstdlib> #include<cstring> #include<iostream> #include<algorithm> using namespace std; int n,Top,Root; struct xcw{int Son[2],tot,x,Num;int &operator [](const int b){return Son[b];}//這個(gè)重載不懂的可以看我的博客:https://blog.csdn.net/qq_41357771/article/details/79405314 }Tre[100005]; int read(){int ret=0;char ch=getchar();bool f=1;for(;!isdigit(ch);ch=getchar()) f^=!(ch^'-');for(; isdigit(ch);ch=getchar()) ret=(ret<<3)+(ret<<1)+ch-'0';return f?ret:-ret; } int Random(){//Random不能寫成random有些編譯器不能過static int seed=703; //seed可以隨便取return seed=int(seed*48271LL%2147483647); } void Change(int x){Tre[x].tot=Tre[Tre[x][0]].tot+Tre[Tre[x][1]].tot+1;} int Turn(int &x,int p){int t=Tre[x][p];Tre[x][p]=Tre[t][!p];Tre[t][!p]=x;Change(x);Change(t);x=t;} void Insert(int &x,int p){if(!x){x=++Top;Tre[Top].tot=1;Tre[Top].x=p;Tre[Top].Num=Random();return;}Tre[x].tot++;if(p<=Tre[x].x){Insert(Tre[x][0],p);if(Tre[Tre[x][0]].Num<Tre[x].Num) Turn(x,0);}else{Insert(Tre[x][1],p);if(Tre[Tre[x][1]].Num<Tre[x].Num) Turn(x,1);} } void Del(int &x,int p){if(Tre[x].x==p){if(Tre[x][0]*Tre[x][1]==0){x=Tre[x][0]+Tre[x][1];return;}if(Tre[Tre[x][0]].Num<Tre[Tre[x][1]].Num){Turn(x,0);Del(Tre[x][1],p);}else{Turn(x,1);Del(Tre[x][0],p);}}else if(Tre[x].x>p) Del(Tre[x][0],p);else Del(Tre[x][1],p);Change(x); } int Fnd(int x,int p){//查找p數(shù)的排名 if(!x) return 1;if(Tre[x].x>=p) return Fnd(Tre[x][0],p);else return Fnd(Tre[x][1],p)+Tre[Tre[x][0]].tot+1; } int Ask(int x,int p){//查找排名為p的數(shù) if(Tre[Tre[x][0]].tot==p-1) return Tre[x].x;if(Tre[Tre[x][0]].tot>=p) return Ask(Tre[x][0],p);else return Ask(Tre[x][1],p-Tre[Tre[x][0]].tot-1); } int FndMax(int x,int p){//找前驅(qū)if(!x) return -(1e9);if(Tre[x].x<p) return max(Tre[x].x,FndMax(Tre[x][1],p));else return FndMax(Tre[x][0],p); } int FndMin(int x,int p){//找后繼 if(!x) return 1e9;if(Tre[x].x>p) return min(Tre[x].x,FndMin(Tre[x][0],p));else return FndMin(Tre[x][1],p); } int main(){ // freopen("prob.in","r",stdin); // freopen("prob.out","w",stdout);n=read();for(int i=1;i<=n;i++){int Opt=read(),x=read();if(Opt==1) Insert(Root,x);elseif(Opt==2) Del(Root,x);elseif(Opt==3) printf("%d\n",Fnd(Root,x));elseif(Opt==4) printf("%d\n",Ask(Root,x));elseif(Opt==5) printf("%d\n",FndMax(Root,x));else printf("%d\n",FndMin(Root,x));}return 0; }

轉(zhuǎn)載于:https://www.cnblogs.com/XSamsara/p/9059421.html

與50位技術(shù)專家面對面20年技術(shù)見證，附贈(zèng)技術(shù)全景圖

總結(jié)

以上是生活随笔為你收集整理的BZOJ 3224: Tyvj 1728 普通平衡树【Treap】的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。