bzoj3224 普通平衡树 splay模板题

题目链接：戳这里

3224: Tyvj 1728 普通平衡树

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 17768  Solved: 7799
[Submit][Status][Discuss]

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]

没什么说的，splay模板题。

代码：

#include<bits/stdc++.h>#define maxn 100005using namespace std;typedef long long LL;int read(){char c;int sum=0,f=1;c=getchar();while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}while(c>='0' && c<='9'){sum=sum*10+c-'0';c=getchar();}return sum*f;}int n,opt,x;int fa[maxn],son[maxn][2],val[maxn],cnt[maxn],size[maxn],sz,root;void clear(int x){son[x][1]=son[x][0]=fa[x]=val[x]=cnt[x]=size[x]=0;}bool getson(int x){return son[fa[x]][1]==x;}void pushup(int x){if(x){size[x]=cnt[x];if(son[x][0]) size[x]+=size[son[x][0]];if(son[x][1]) size[x]+=size[son[x][1]];}}void rotate(int x){int y=fa[x],z=fa[y],s=getson(x);son[y][s]=son[x][s^1];fa[son[y][s]]=y;fa[y]=x;son[x][s^1]=y;fa[x]=z;if(z)son[z][son[z][1]==y]=x;pushup(y);pushup(x);}void splay(int x,int k){for(int y;(y=fa[x])!=k;rotate(x))if(fa[y]!=k)rotate((getson(x)==getson(y))?y:x);if(!k) root=x;}void insert(int x){if(root==0){sz++;root=sz;son[sz][0]=son[sz][1]=fa[sz]=0;val[sz]=x;size[sz]=cnt[sz]=1;return;}int now=root,f=0;while(1){if(val[now]==x){cnt[now]++;pushup(now);pushup(f);splay(now,0);break;}f=now;now=son[now][val[now]<x];if(now==0){sz++;son[sz][0]=son[sz][1]=0;fa[sz]=f;val[sz]=x;size[sz]=cnt[sz]=1;son[f][val[f]<x]=sz;pushup(f);splay(sz,0);break;}}}int find(int x){int ans=0,now=root;while(1){if(x<val[now])now=son[now][0];else{ans+=(son[now][0]?size[son[now][0]]:0);if(x==val[now]){splay(now,0);return ans+1; }ans+=cnt[now];now=son[now][1];}}}int findx(int x){int now=root;while(1){if(son[now][0] && x<=size[son[now][0]])now=son[now][0];else{int temp=(son[now][0]?size[son[now][0]]:0)+cnt[now];if(x<=temp)return val[now];x-=temp;now=son[now][1];}}}int pre(){int now=son[root][0];while(son[now][1])now=son[now][1];      return now;}int nex(){int now=son[root][1];while(son[now][0])now=son[now][0];return now;}void del(int x){find(x);if(cnt[root]>1){cnt[root]--;return;}if(!son[root][0] && !son[root][1]){clear(root);root=0;return;}int oldroot=root;if(!son[root][0]){root=son[root][1];fa[root]=0;clear(oldroot);return;}else if(!son[root][1]){root=son[root][0];fa[root]=0;clear(oldroot);return;}splay(pre(),0);fa[son[oldroot][1]]=root;son[root][1]=son[oldroot][1];clear(oldroot);pushup(root);}int main(){n=read();for(int i=1;i<=n;i++){opt=read();x=read();switch (opt){case 1:insert(x);break;case 2:del(x);break;case 3:printf("%d\n",find(x));break;case 4:printf("%d\n",findx(x));break; case 5:insert(x);printf("%d\n",val[pre()]);del(x);break;              case 6:insert(x);printf("%d\n",val[nex()]);del(x);break;  }}return 0;}