splay版

指针是个好东西

不过就是得判空

还有别忘传引用(其实应该都传引用)

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
int inf=0x7fffffff;
struct node* nil;
struct node
{
	int num;
	int val;
	int size;
	node* ch[2];
	node (int v) :val(v)
	{
		size=1;
		num=1;
		ch[0]=ch[1]=nil;
	}
	void sum()
	{
		size=num;
		if(ch[0]!=nil)
			size+=ch[0]->size;
		if(ch[1]!=nil)
			size+=ch[1]->size;
		return ;
	}
	int cmp(int v)
	{
		if(val==v)
			return -1;
		return (val>v ? 0 : 1);
	}
	int cmpkth(int k)
	{
		int s=( ch[0]==nil ? 0 : ch[0]->size );
		if(k>s&&k<=s+num)
			return -1;
		if(k<=s)
			return 0;
		else
			return 1;
	}
};
node* root;
void visit(node* x)
{
	if(x==nil)
		return ;
	visit(x->ch[0]);
	printf("%d ",x->val);
	visit(x->ch[1]);
	return ;
}
void rotato(node* &x,int base)
{
	node* k=x->ch[base^1];
	x->ch[base^1]=k->ch[base];
	k->ch[base]=x;
	x->sum();
	k->sum();
	x=k;
}
void splay(node* &x,int v)
{
	int d=x->cmp(v);
	if(d!=-1&&x->ch[d]!=nil)
	{
		int d2=x->ch[d]->cmp(v);
		if(d2!=-1&&x->ch[d]->ch[d2]!=nil)
		{
			splay(x->ch[d]->ch[d2],v);
			if(d==d2)
				rotato(x,d2^1),rotato(x,d^1);
			else
				rotato(x->ch[d],d2^1),rotato(x,d^1);
		}
		else
			rotato(x,d^1);
	}
}
void splaykth(node* &x,int k)
{
	int d=x->cmpkth(k);
	if(d!=-1)
	{
		if(d==1)
			k=k-x->ch[0]->size-x->num;
		int d2=x->ch[d]->cmpkth(k);
		if(d2!=-1)
		{
			int k2=(d2==1 ? k-x->ch[d]->ch[0]->size-x->ch[d]->num : k);
			splaykth(x->ch[d]->ch[d2],k2);
			if(d==d2)
				rotato(x,d2^1),rotato(x,d^1);
			else
				rotato(x->ch[d],d2^1),rotato(x,d^1);
		}
		else
			rotato(x,d^1);
	}
	return ;
}
void pre(node* x,int val,int &ans)
{
	if(x==nil)
		return ;
	if(x->val<val)
	{
		if(x->val>ans)
			ans=x->val;
		if(x->ch[1]!=nil)
			pre(x->ch[1],val,ans);
	}
	else
		if(x->val>=val&&x->ch[0]!=nil)
			pre(x->ch[0],val,ans);
}
void nxt(node* x,int val,int &ans)
{
	if(x==nil)
		return ;
	if(x->val>val)
	{
		if(x->val<ans)
			ans=x->val;
		if(x->ch[0]!=nil)
			nxt(x->ch[0],val,ans);
	}
	else
		if(x->val<=val&&x->ch[1]!=nil)
			nxt(x->ch[1],val,ans);
}
int find(node* &x,int val)
{
	splay(x,val);
	return x->ch[0]->size+1;
}
int kth(node* &x,int k)
{
	splaykth(x,k);
	return x->val;
}
node *spilt(node* &x,int val)
{
	if(x==nil)
		return nil;
	splay(x,val);
	node* t1;
	node* t2;
	if(x->val<=val)
		t1=x,t2=x->ch[1],t1->ch[1]=nil;
	else
		t2=x,t1=x->ch[0],t2->ch[0]=nil;
	x->sum();
	x=t1;
	return t2;
}
void merge(node* &t1,node* &t2)
{
	if(t1==nil)
		swap(t1,t2);
	splay(t1,inf);
	t1->ch[1]=t2;
	t2=nil;
	t1->sum();
}
void insert(node* &x,int val)
{
	//visit(x);printf("
");
	node* t2=spilt(x,val);
	//visit(x);printf("
");
	if(x!=nil&&x->val==val)
	{
		x->num+=1;
		x->sum();
	}
	else
	{
		node* nw=new node(val);
		merge(x,nw);
	}
	merge(x,t2);
	//visit(root);printf("
");
}
void erase(node* &x,int val)
{
	node* t2=spilt(x,val);
	x->num-=1;
	if(x->num==0)
	{
		node* t3=x;
		x=x->ch[0];
		delete t3;
	}
	merge(x,t2);
}
int read()
{
	int s=0,f=1;
	char in=getchar();
	while(in<'0'||in>'9')
	{
		if(in=='-')
			f=-1;
		in=getchar();
	}
	while(in>='0'&&in<='9')
	{
		s=(s<<1)+(s<<3)+in-'0';
		in=getchar();
	}
	return s*f;
}
int main()
{
	int n=read();
	int a,b;
	int ans;
	nil=new node(0);
	root=nil->ch[0]=nil->ch[1]=nil;
	nil->size=nil->num=0;
	for(int i=1;i<=n;i++)
	{
		a=read();
		b=read();
		switch(a)
		{
			case 1: insert(root,b);break;
			case 2:	erase(root,b);break;
			case 3: printf("%d
",find(root,b));break;
			case 4: printf("%d
",kth(root,b));break;
			case 5:	insert(root,b);ans=-0x7fffffff;pre(root,b,ans);printf("%d
",ans);erase(root,b);break;
			case 6: insert(root,b);ans=0x7fffffff;nxt(root,b,ans);printf("%d
",ans);erase(root,b);break;
			default: break;
		}
	}
	return 0;
}
原文地址:https://www.cnblogs.com/Lance1ot/p/8947372.html