伸展树模板HYSBZ - 3224

题目链接

 

 

#include<cstdio>
#include<iostream>
using namespace std;

#define maxn 100005

int sz,root,ch[maxn][2],f[maxn],cnt[maxn],key[maxn],size[maxn];

inline void clear(int x){
	ch[x][0]=ch[x][1]=f[x]=size[x]=cnt[x]=key[x]=0;
}
inline bool get(int x){
	return ch[f[x]][1]==x;
}
inline void update(int x){
	if (x){
		size[x]=cnt[x];
		if (ch[x][0]) size[x]+=size[ch[x][0]];
		if (ch[x][1]) size[x]+=size[ch[x][1]];
	}
}
inline void rotate(int x){
	int old=f[x];int oldf=f[old];int now=get(x);
	ch[old][now]=ch[x][now^1];f[ch[old][now]]=old;
	ch[x][now^1]=old;
	f[x]=oldf;f[old]=x;
	if(oldf){
		ch[oldf][ch[oldf][1]==old]=x;
	}
	update(old);update(x);
}
inline void splay(int x){
	for (int fa;fa=f[x];rotate(x))
	  if (f[fa])
	    rotate((get(x)==get(fa))?fa:x);
	root=x;
}
inline void insert(int x){
	if(root==0){
		sz++;
		root=sz;
		f[sz]=ch[sz][0]=ch[sz][1]=0;
		cnt[sz]=size[sz]=1;
		key[sz]=x;return ;
	}
	int now=root,fa=0;
	while(1){
		if(x==key[now]){
			cnt[now]++;update(now);update(fa);splay(now);break;
		}
		fa=now;
		now=ch[now][x>key[now]];
		if (now==0){
			sz++;
			ch[sz][0]=ch[sz][1]=0;
			f[sz]=fa;
			size[sz]=cnt[sz]=1;
			ch[fa][key[fa]<x]=sz;
			key[sz]=x;
			update(fa);
			splay(sz);
			break;
		}

	}
}

inline int find(int x){
	int now=root,ans=0;
	while(1){
		if(key[now]>x){
			now=ch[now][0];
		}
		else{
			ans+=(ch[now][0]?size[ch[now][0]]:0);
			if(key[now]==x) {
				splay(now);
				return ans+1;
			}
			ans+=cnt[now];
			now=ch[now][1];
			
		}
	}
}
inline int findx(int x){
	int now=root;
	while(1){
		if(size[ch[now][0]]>=x&&ch[now][0]){
			now=ch[now][0];
		}
		else{
			int temp=(ch[now][0]?size[ch[now][0]]:0)+cnt[now];
			if(temp>=x){
				return key[now];
			}
			x-=temp;
			now=ch[now][1];
		}
			
	}
}
inline int pre(){
	int now=ch[root][0];
	while (ch[now][1]) now=ch[now][1];
	return now;
}
inline int next(){
	int now=ch[root][1];
	while (ch[now][0]) now=ch[now][0];
	return now;
}
inline void del(int x){
	int where=find(x);
	if(cnt[root]>1){
		cnt[root]--;
		update(root);
		return ;
	}
	if(!ch[root][0]&&!ch[root][1]){
		clear(root);root=0;return ;
	}
	if(!ch[root][0]){
		int oldroot=root;
		root=ch[root][1];
		f[root]=0;
		clear(oldroot);
		return ;
	}
	if(!ch[root][1]){
		int oldroot=root;
		root=ch[root][0];
		f[root]=0;
		clear(oldroot);
		return ;
	}
	int oldroot=root;
	int pres=pre();
	splay(pres);
	ch[root][1]=ch[oldroot][1];
	f[ch[root][1]]=root;
	clear(oldroot);
	update(root);
	
}

int main(){
	int n,opt,x;
	scanf("%d",&n);
	for (int i=1;i<=n;++i){
		scanf("%d%d",&opt,&x);
		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",key[pre()]); del(x); break;
			case 6: insert(x); printf("%d\n",key[next()]); del(x); break;
		}
	}
	return 0;
}