bzoj 3224: Tyvj 1728 普通平衡树 （Splay模板）

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=3224

题意：

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：

1. 插入x数

2. 删除x数(若有多个相同的数，因只删除一个)

3. 查询x数的排名(若有多个相同的数，因输出最小的排名)

4. 查询排名为x的数

5. 求x的前驱(前驱定义为小于x，且最大的数)

6. 求x的后继(后继定义为大于x，且最小的数)

分析：

平衡树的入门题，可以当做一般平衡树操作的模板了~~

这题我是WA了好多次，我是用http://blog.csdn.net/crazy_ac/article/details/8034190这个大神的模板，不过总是WA，改了一些地方终于AC了，前几天可是耗费了我一天的功夫啊！，还是太菜了QAQ

代码：

#include<cstdio>
#include<cstdlib>
const int inf  = ~0u>>2;
#define L ch[x][0]
#define R ch[x][1]
#define KT (ch[ ch[rt][1] ][0])
const int maxn = 1e5+9;
int lim;
struct SplayTree {
int sz[maxn];
int ch[maxn][2];
int pre[maxn];
int rt,top;
inline void up(int x) {
sz[x]  = cnt[x]  + sz[ L ] + sz[ R ];
}
inline void Rotate(int x,int f) {
int y=pre[x];
ch[y][!f] = ch[x][f];
pre[ ch[x][f] ] = y;
pre[x] = pre[y];
if(pre[x]) ch[ pre[y] ][ ch[pre[y]][1] == y ] =x;
ch[x][f] = y;
pre[y] = x;
up(y);
}
inline void Splay(int x,int goal) { //将x旋转到goal的下面
while(pre[x] != goal) {
if(pre[pre[x]] == goal) Rotate(x , ch[pre[x]][0] == x);
else   {
int y=pre[x],z=pre[y];
int f = (ch[z][0]==y);
if(ch[y][f] == x) Rotate(x,!f),Rotate(x,f);
else Rotate(y,f),Rotate(x,f);
}
}
up(x);
if(goal==0) rt=x;
}
inline void RTO(int k,int goal) { //将第k位数旋转到goal的下面
int x=rt;
while(sz[ L ] != k-1) {
if(k < sz[ L ]+1) x=L;
else {
k-=(sz[ L ]+1);
x = R;
}
}
Splay(x,goal);
}

inline void Newnode(int &x,int c,int f) {
x=++top;
L = R = 0;
pre[x] = f;
sz[x]=1;
cnt[x]=1;
val[x] = c;
}
inline void init() {
ch[0][0]=ch[0][1]=pre[0]=sz[0]=0;
rt=top=0;
cnt[0]=0;
}
inline void Insert(int &x,int key,int f) {
if(!x) {
Newnode(x,key,f);
Splay(x,0);//注意插入完成后splay
return ;
}
if(key==val[x]) {
cnt[x]++;
sz[x]++;
Splay(x,0);//注意插入完成后splay
return ;
} else if(key<val[x]) {
Insert(L,key,x);
} else {
Insert(R,key,x);
}
up(x);
}
void Del_root() { //删除根节点
int t=rt;
if(ch[rt][1]) {
rt=ch[rt][1];
RTO(1,0);
ch[rt][0]=ch[t][0];
if(ch[rt][0]) pre[ch[rt][0]]=rt;
} else rt=ch[rt][0];
pre[rt]=0;
up(rt);
}
void findpre(int x,int key,int &ans) { //找前驱节点
ans=-inf;
while(x) {
if(val[x]<key) {
if(val[x]>ans)ans=val[x];
x=R;
} else x=L;
}
}
void findsucc(int x,int key,int &ans) { //找后继节点
ans=inf;
while(x) {
if(val[x]>key) {
if(val[x]<ans)ans=val[x];
//  ans=min(ans,val[x]);
x=L;
} else x=R;
}
}
inline int find_kth(int x,int k) { //第k小的数
if(k<sz[L]+1) {
return find_kth(L,k);
} else if(k > sz[ L ] + cnt[x] )
return find_kth(R,k-sz[L]-cnt[x]);
else {
Splay(x,0);
return val[x];
}
}
int find(int key) {
int x=rt;
while(x&&val[x]!=key)
x=ch[x][key>val[x]];
if(x)Splay(x,0); //0是没找到
return x;
}
int find(int x,int key) {
//if(!x) return 0;
while(x&&val[x]!=key)x=ch[x][key>val[x]];
return x;
}

int rank(int key) {
int x = find(key);
//Splay(x,rt);
return sz[ch[rt][0]]+1;
}
//确定key的排名
int get_rank(int x) {
int *key=val;
int r=rt,temp=0;

while(ch[r][key[r]<=x]) {
if(key[r]==x) break;
if(key[r]<=x) temp+=sz[r]-sz[ch[r][1]];
r=ch[r][key[r]<=x];
}
return sz[ch[r][0]]+1+temp;
}
int getrank(int x,int key,int cur) { //cur:当前已知比要求元素（key）小的数的个数
if(key == val[x])
return sz[L] + cur + 1;
else if(key < val[x])
return getrank(L,key,cur);
else
return getrank(R,key,cur+sz[L]+cnt[rt]);
}

void del(int &x,int f) { //删除小于lim的所有的数所在的节点
if(!x) return ;
if(val[x]>=lim) {
del(L,x);
} else {
x=R;
pre[x]=f;
if(f==0)  rt=x;
del(x,f);
}
if(x)  up(x);
}

void Delete(int key) {
int node=find(rt,key);
Splay(node,0);
cnt[rt]--;
if(cnt[rt]<=0)Del_root();
}

void solve(int n) {
int x,op,ans,tot=0;
for(int i=0; i<n; i++) {
scanf("%d%d",&op,&x);
if(op==1) {
Insert(rt,x,0);
continue;
} else if(op==2) {
Delete(x);
continue;
} else if(op==3) {
ans=rank(x);
} else if(op==4) {
ans=find_kth(rt,x);
} else if(op==5) {
findpre(rt,x,ans);
} else if(op==6) {
findsucc(rt,x,ans);
}
printf("%d\n",ans);
}
}
int cnt[maxn];
int val[maxn];
int lim;
} spt;

int main() {
int n;
while(~scanf("%d",&n)) {
spt.init();
spt.solve(n);
}
return 0;
}