3224: Tyvj 1728 普通平衡樹

Time Limit: 10 Sec Memory Limit: 128 MB

Description

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

Input

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

Output

對于操作3,4,5,6每行輸出一個數，表示對應答案

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的數據范圍：n<=100000
2.每個數的數據范圍：[-2e9,2e9]

題解

裸的平衡樹，因為Treap好寫，所以就寫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];}//這個重載不懂的可以看我的博客: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數的排名 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的數 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){//找前驅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; }

轉載于:https://www.cnblogs.com/XSamsara/p/9059421.html

與50位技術專家面對面20年技術見證，附贈技術全景圖

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。