24分 初学Treap 求大佬帮忙调调 QAQ
查看原帖
24分 初学Treap 求大佬帮忙调调 QAQ
182792
Jie_Rans楼主2022/1/10 02:31
#include<bits/stdc++.h>
#define int long long 
using namespace std;
const int N=1e5+10,INF=1e9+10;
int n,root,tot;
struct Node
{
	int l,r,val,size,cnt,dat;
}t[N];
namespace BST
{
	//×óÐý 
	void left_cir(int &p)
	{
		int q=t[p].r;
		t[p].r=t[q].l;
		t[q].l=p;
		p=q;
	}
	//ÓÒÐý
	void right_cir(int &p)
	{
		int q=t[p].l;
		t[p].l=t[q].r;
		t[q].r=p;
		p=q;
	}  
	int addpoint(int val)
	{
		t[++tot].val=val;
		t[tot].dat=rand();
		t[tot].cnt=t[tot].size=1;
		return tot;
	}
    void update(int p)
	{
		t[p].size=t[t[p].l].size+t[t[p].r].size+t[p].cnt;
	}
	void build()
	{
		srand(time(0));
		addpoint(-INF);
		addpoint(INF);
		root=1;
		t[1].r=2;
		update(root);
	}
	//²Ù×÷1 ²åÈë 
	void insert(int &p,int val)
	{
	//	cout<<p<<" "<<val<<endl;
		if(p==0)
		{
			p=addpoint(val);
			return ;
		}
		if(val==t[p].val) 
		{
			t[p].cnt++;
			update(p);
			return ;
		}
		if(val<t[p].val)
		{
			insert(t[p].l,val);
			if(t[p].dat<t[t[p].l].dat) right_cir(p);	
		}
		if(val>t[p].val) 
		{
			insert(t[p].r,val);
			if(t[p].dat<t[t[p].r].dat) left_cir(p);
		}
	}
	//²Ù×÷2 ɾ³ý 
	void dele(int &p,int val)
	{
		if(!p) return ;
		if(t[p].val==val)
		{
			if(t[p].cnt>1)
			{
				t[p].cnt--;
				update(p);
				return ;
			}
			if(t[p].l || t[p].r)
			{
				if(!t[p].r || t[t[p].l].dat>t[t[p].r].dat) 
				{
					right_cir(p);
					dele(t[p].r,val);
				}
				else 
				{
					left_cir(p);
					dele(t[p].l,val);
				}
				update(p);
			}
			else 
				p=0;
			return ;
		}
		val<t[p].val?dele(t[p].l,val):dele(t[p].r,val);
		update(p);
	}
	//²Ù×÷3 ÇóÅÅÃû 
	int getrank(int p,int val)
	{
		if(!p) return 0;
		if(val==t[p].val) return t[t[p].l].size+1;
		if(val<t[p].val) return getrank(t[p].l,val);
		return getrank(t[p].r,val)+t[t[p].l].size+t[p].cnt;
	}
	//²Ù×÷4 ÇóÊýÖµ 
	int getval(int p,int rank)
	{
		if(!p) return INF;
		if(t[t[p].l].size>=rank) return getval(t[p].l,rank);
		if(t[t[p].l].size+t[p].cnt>=rank) return t[p].val;
		return getval(t[p].r,rank-t[t[p].l].size-t[p].cnt);
	}
	//²Ù×÷6 Çóºó׺
	int getnxt(int val)
	{
		int ans=2,p=root;
		while(p)
		{
			if(val==t[p].val)
			{
				if(t[p].r>0)
				{
					p=t[p].r;
					while(t[p].l>0) p=t[p].l;
					ans=p;
				}
				break;
			}
			if(t[p].val>val && t[p].val<t[ans].val) ans=p;
			p=val<t[p].val?t[p].l:t[p].r;
		}
		return t[ans].val;
	} 
	//²Ù×÷5 ÇóÇ°Çý
	int getpre(int val)
	{
		int ans=1,p=root;
		while(p)
		{
			if(val==t[p].val)
			{
				if(t[p].l>0)
				{
					p=t[p].l;
					while(t[p].r>0) p=t[p].r;
					ans=p;
				}
				break;
			}
			if(t[p].val<val && t[p].val>t[ans].val) ans=p;
			p=val<t[p].val?t[p].l:t[p].r;
		}
		return t[ans].val;
	 } 
};
int read()
{
	char ch=getchar();
	int x=0,f=1;
	while(ch<'0' || ch>'9')
	{
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(ch>='0' && ch<='9')
	{
		x=x*10+ch-'0';
		ch=getchar();
	}
	return x*f;
}
signed main()
{
	freopen("ljian_Cheng.in","r",stdin);
	freopen("ljian_Cheng.out","w",stdout);
	BST::build();
	n=read();
	while(n--)
	{
		int opt=read(),x=read();
		switch(opt)
		{
			case 1:
				BST::insert(root,x);
				break;
			case 2:
				BST::dele(root,x);
				break;
			case 3:
				cout<<BST::getrank(root,x)-1<<endl;
				break;
			case 4:
				cout<<BST::getval(root,x+1)<<endl;
				break;
			case 5:
				cout<<BST::getpre(x)<<endl;
				break;
			case 6:
				cout<<BST::getnxt(x)<<endl;
				break;
		}
	}
	return 0;
}
2022/1/10 02:31
加载中...