记录编号 213594 评测结果 AAAAAAAAAA
题目名称 [Tyvj 1728]普通平衡树 最终得分 100
用户昵称 Gravatarstdafx.h 是否通过 通过
代码语言 C++ 运行时间 0.175 s
提交时间 2015-12-12 19:48:08 内存使用 2.58 MiB
显示代码纯文本
//ScapeGoat_Tree
#define MAXN 100010UL

#define INF 0x4fffffff

#define fa(x) (tree[x].fa)
#define ls(x) (tree[x].l)
#define rs(x) (tree[x].r)
#define val(x) (tree[x].key)
#define bs(x) (tree[x].bs)

#include <cmath>
#include <cstdio>

struct SGT_Node{int fa,l,r,key,bs;};
const double Alpha=0.75,logAlpha=log(1.0/Alpha);
SGT_Node tree[MAXN];
int n,root,cnt,dfs_clock,lis[MAXN];

inline int in(){
	int x=0,c=getchar();
	bool flag=false;
	while(c<'0'||c>'9'){
		if(c=='-') flag=true;
		c=getchar();
	}
	for(;c>='0'&&c<='9';c=getchar()) x=(x<<1)+(x<<3)+(c^48);
	if(flag) return -x;
	return x;
}

inline bool Balanced(int p){
	return bs(p)*Alpha>=bs(ls(p))&&bs(p)*Alpha>=bs(rs(p));
}

inline int MAX(int a,int b){
	return a>b?a:b;
}

inline int MIN(int a,int b){
	return a<b?a:b;
}

void dfs(int p){
	if(ls(p)) dfs(ls(p));
	lis[++dfs_clock]=p;
	if(rs(p)) dfs(rs(p));
	return;
}

int Build(int l,int r){
	if(l>r) return 0;
	int mid=(l+r)>>1,p=lis[mid];
	fa(ls(p)=Build(l,mid-1))=p;
	fa(rs(p)=Build(mid+1,r))=p;
	bs(p)=bs(ls(p))+bs(rs(p))+1;
	return p;
}

inline void ReBuild(int p){
	dfs_clock=0;dfs(p);
	int f=fa(p),res=(rs(fa(p))==p),new_root=Build(1,dfs_clock);
	if(!res) fa(ls(f)=new_root)=f;
	else fa(rs(f)=new_root)=f;
	if(p==root) root=new_root;
	return;
}

inline int Rank(int p){
	int Ans=0,temp=root;
	while(temp){
		if(val(temp)<p) Ans+=bs(ls(temp))+1,temp=rs(temp);
		else temp=ls(temp);
	}
	return Ans;
}

inline int GetKth(int p){
	int temp=root;
	while(temp){
		if(bs(ls(temp))+1==p) return temp;
		else if(bs(ls(temp))>=p) temp=ls(temp);
		else p-=bs(ls(temp))+1,temp=rs(temp);
	}
	return temp;
}

inline int Find(int p){
	int temp=root;
	while(temp){
		if(val(temp)==p) return temp;
		else if(val(temp)<p) temp=rs(temp);
		else temp=ls(temp);
	}
	return temp;
}

inline int Pre(int p){
	int temp=root,Ans=-INF;
	while(temp){
		if(val(temp)<p) Ans=MAX(Ans,val(temp)),temp=rs(temp);
		else temp=ls(temp);
	}
	return Ans;
}

inline int Next(int p){
	int temp=root,Ans=INF;
	while(temp){
		if(val(temp)>p) Ans=MIN(Ans,val(temp)),temp=ls(temp);
		else temp=rs(temp);
	}
	return Ans;
}

inline void Delete(int p){
	if(ls(p)&&rs(p)){
		int temp=ls(p);
		while(rs(temp)) temp=rs(temp);
		val(p)=val(temp);
		p=temp;
	}
	int son=ls(p)?ls(p):rs(p),res=(rs(fa(p))==p);
	if(res) fa(rs(fa(p))=son)=fa(p);
	else fa(ls(fa(p))=son)=fa(p);
	for(int i=fa(p);i;i=fa(i)) bs(i)--;
	if(p==root) root=son;
	return;
}

inline void Insert(int p){
	int temp=root,depth=0;
	++cnt;bs(cnt)=1,val(cnt)=p;
	while(true){
		bs(temp)++;
		if(p<val(temp)){
			if(ls(temp)) temp=ls(temp);
			else{fa(ls(temp)=cnt)=temp;break;}
		}
		else{
			if(rs(temp)) temp=rs(temp);
			else{fa(rs(temp)=cnt)=temp;break;}
		}
		depth++;
	}
	if(depth>log(cnt)/logAlpha){
		int goat=0;
		for(temp=cnt;temp;temp=fa(temp)) if(!Balanced(temp)) goat=temp;
		if(goat) ReBuild(goat);
	}
	return;
}

int main(){
	freopen("phs.in","r",stdin);
	freopen("phs.out","w",stdout);
	n=in();root=1,cnt=2;
	val(1)=-INF,rs(1)=2,bs(1)=2;
	val(2)=+INF,fa(2)=1,bs(2)=1;
	for(int i=1,op,p;i<=n;i++){
		op=in(),p=in();
		if(op==1) Insert(p);
		else if(op==2) Delete(Find(p));
		else if(op==3) printf("%d\n",Rank(p));
		else if(op==4) printf("%d\n",val(GetKth(p+1)));
		else if(op==5) printf("%d\n",Pre(p));
		else if(op==6) printf("%d\n",Next(p));
	}
	return 0;
}