LOJ104 普通平衡树【Splay】

传送门

---

SOL:

splay板子

---

#include
using namespace std;
#define sf scanf
#define pf printf
const int maxn=1e5+10;
int son[maxn][2],siz[maxn],sum[maxn],fa[maxn],val[maxn],n,root=0,tot=0;
inline void pushup(int u){siz[u]=siz[son[u][0]]+siz[son[u][1]]+sum[u];}
inline bool check(int x){return x==son[fa[x]][1];}
inline void rotate(int x){
	int y=fa[x],z=fa[y],d=check(x),d2=check(y);
	son[y][d]=son[x][d^1];fa[son[y][d]]=y;
	son[x][d^1]=y;fa[y]=x;
	fa[x]=z;if(z)son[z][d2]=x;
	pushup(y);pushup(x);
}
inline void splay(int u,int tar){
	while(fa[u]!=tar){
		if(fa[fa[u]]!=tar){
			if(check(u)==check(fa[u]))rotate(fa[u]);
			else rotate(u);
		}
		rotate(u);
	}
	if(!tar)root=u;
}
inline void insert(int k){
	int u=root,v=0,d;
	while(u){
		++siz[v=u];
		if(k==val[u]){
			++sum[u];return;
		}
		if(k<val[u])u=son[u][0],d=0;
		else u=son[u][1],d=1;
	}
	u=++tot;fa[u]=v;siz[u]=sum[u]=1;val[u]=k;
	if(v)son[v][d]=u;
	splay(u,0);
}
//按数值找  0：位置 1：排名
inline int rank(int k,bool f){
	int u=root,ret=0;
	while(u){
		if(val[u]==k)return f ? ret+siz[son[u][0]]+1 : u;
		if(val[u]>k)u=son[u][0];
		else ret+=siz[son[u][0]]+sum[u],u=son[u][1];
	}
	return f ? ret : 0;
} 
//按 排名找  
inline int find(int k){
	int u=root;
	while(u){
		if(siz[son[u][0]]>=k) u= son[u][0];
		else if( siz[son[u][0]]+sum[u]>=k) return val[u];
		else k-=(siz[son[u][0]]+sum[u]),u=son[u][1];
	}
	return val[u];
}
inline void delet(int k){
	int u=rank(k,0);
	splay(u,0);
	if(sum[u]>1){
		--siz[u];--sum[u];return;
	}
	int l=son[u][0],r=son[u][1];
	fa[l]=fa[r]=0;
	if(!l||!r)root=l+r;
	else{
		int w=l;
		while(son[w][1])w=son[w][1];
		splay(w,0);
		son[w][1]=r;fa[r]=w;
		pushup(w);
	}
}
//0: 前驱  1：后继 
inline int aft(int k){
	int u=root,ret=0;
	while(u){
		if(val[u]>k)ret=val[u],u=son[u][0];
		else u=son[u][1];
	}
	return ret;
}
inline int pre(int k){
	int u=root,ret=0;
	while(u){
		if(val[u]<k)ret=val[u],u=son[u][1];
		else u=son[u][0];
	}
	return ret;
}
signed main(){
	sf("%d",&n);
	while(n--){
		int f,x;sf("%d%d",&f,&x);
		if(f==1)insert(x);
		if(f==2)delet(x);
		if(f==3)cout<<rank(x,1)<<'\n';
		if(f==4)cout<<find(x)<<'\n';
		if(f==5)cout<<pre(x)<<'\n';
		if(f==6)cout<<aft(x)<<'\n';
	}	
	return 0;
}