[luogu3369] 普通平衡树（splay模板）

题目描述

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

1.插入 xx 数
2.删除 xx 数(若有多个相同的数，因只删除一个)
3.查询 xx 数的排名(排名定义为比当前数小的数的个数 +1+1 。若有多个相同的数，因输出最小的排名)
4.查询排名为 xx 的数
5.求 xx 的前驱(前驱定义为小于 xx ，且最大的数)
6.求 xx 的后继(后继定义为大于 xx ，且最小的数)

输入输出格式

输入格式：
第一行为 nn ，表示操作的个数,下面 nn 行每行有两个数 optopt 和 xx ， optopt 表示操作的序号( 1 \leq opt \leq 6 1≤opt≤6 )

输出格式：
对于操作 3,4,5,63,4,5,6 每行输出一个数，表示对应答案

输入输出样例

输入样例#1：
10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598
输出样例#1：
106465
84185
492737

说明

code:

//By Menteur_Hxy
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define M(a,b) memset(a,(b),sizeof(a))
#define F(i,a,b) for(register int i=(a);i<=(b);i++)
#define C(i,a,b) for(register int i=(b);i>=(a);i--)
#define E(i,u) for(register int i=head[u];i;i=nex[i])
using namespace std;

inline LL rd() {
	LL x=0,fla=1; char c=' ';
	while(c>'9'|| c<'0') {if(c=='-') fla=-fla; c=getchar();}
	while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
	return x*fla;
}

inline void out(LL x){
    int a[25],wei=0;
    if(x<0) putchar('-'),x=-x;
    for(;x;x/=10) a[++wei]=x%10;
    if(wei==0){ puts("0"); return;}
    for(int j=wei;j>=1;--j) putchar('0'+a[j]);
    putchar('\n');
}
/*--splay_begin--*/ 
#define check(x) (x==nd[fa[x]][1])
#define clear(x) key[x]=cnt[x]=siz[x]=fa[x]=nd[x][1]=nd[x][0]=0
#define update(x) siz[x]=siz[nd[x][1]]+siz[nd[x][0]]+cnt[x]
const int N=100005;
int key[N],cnt[N],siz[N],fa[N],nd[N][2];
int size,root;

inline int newnode(int x) {key[++size]=x,cnt[size]=siz[size]=1;return size;}
inline void rotate(int x) { int old=fa[x],oldf=fa[old];
	bool wh=check(x);
	fa[nd[x][wh^1]]=old; nd[old][wh]=nd[x][wh^1];
	nd[x][wh^1]=old; fa[old]=x; fa[x]=oldf;
	if(oldf) nd[oldf][nd[oldf][1]==old]=x;
	update(old);update(x);
}
inline void splay(int x) {
	for(int f=fa[x];f;rotate(x),f=fa[x])
		if(fa[f]) rotate(check(f)==check(x)?f:x);
	root=x;
}
inline void insert(int x) {
	if(!root) {root=newnode(x);return ;}
	int now=root,f=0;
	while(1) {
		if(key[now]==x) {cnt[now]++;update(now);update(f);splay(now);return ;}
		f=now,now=nd[now][x>key[now]];
		if(!now) {now=newnode(x);fa[now]=f;nd[f][x>key[f]]=now;update(f);splay(now);return ;}
	}
}
inline int rk(int x) { int now=root,ans=0;
	while(1) {
		if(x<key[now]) now=nd[now][0];
		else {
			ans+=siz[nd[now][0]];
			if(key[now]==x) return splay(now),ans+1;
			ans+=cnt[now]; now=nd[now][1];
		}
	}
}
inline int kth(int x) { int now=root;
	while(1) {
		if(nd[now][0]&&x<=siz[nd[now][0]]) now=nd[now][0];
		else {
			x-=siz[nd[now][0]]+cnt[now];
			if(x<=0) return now;
			now=nd[now][1];
		}
	}
}
inline int prenxt(int now) { int wh=check(now);
	while(nd[now][wh^1]) now=nd[now][wh^1];
	return now;
}
inline void del(int x) {
	rk(x);
	if(cnt[root]>1) {cnt[root]--;update(root);return ;}
	if(!nd[root][0]&&!nd[root][1]) {clear(root);root=0;return ;}
	if(!nd[root][0]) {int old=root;root=nd[root][1];fa[root]=0;clear(old);return ;}
	if(!nd[root][1]) {int old=root;root=nd[root][0];fa[root]=0;clear(old);return ;}
	int old=root;
	splay(prenxt(nd[root][0]));
	nd[root][1]=nd[old][1];
	fa[nd[old][1]]=root;
	clear(old);
	update(root);
	return ; 
}
/*--splay_end--*/
int main() {
    int n=rd();
    for(int i=1;i<=n;i++) { int opt=rd(),x=rd();
        switch(opt) { 
            case 1:insert(x);break;
            case 2:del(x);break;
            case 3:out(rk(x));break;
            case 4:out(key[kth(x)]);break;
            case 5:insert(x);out(key[prenxt(nd[root][0])]);del(x);break;
            case 6:insert(x);out(key[prenxt(nd[root][1])]);del(x);break;   
        }
    }
    return 0;
}