P6136 【模板】普通平衡树(数据加强版)

题目

题目

思路

过了板子再做加强岂不妙哉
注意,本题要开long long.
出题人:本题输入数据较大,请使用较快的读入方式。
做题人:Treap+cin可AC,弄啥嘞
AC记录(请出题人加强数据)
code:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
using namespace std;
struct Treap{
    
    
	long long l,r;//左右子编号 
	long long v,d;//v权值(真),d随机值(旋转用) 
	long long cnt,size;//cnt重复个数,size子树大小 
} a[1100010];
long long tot,root=1,n,inf=1e31,rank1,v;
long long New()//新建节点 
{
    
    
	a[++tot].v=v;
	a[tot].d=rand();
	a[tot].cnt=1;
	a[tot].size=1;
	return tot;
}
void Up(long long x)
{
    
    
	a[x].size=a[a[x].l].size+a[a[x].r].size+a[x].cnt;
	return; 
}
void Build()
{
    
    
	v=-inf;
	root=New();
	v=inf;
	a[root].r=New();
	//避免越界
	Up(root);
	return;
}
long long GRBV(long long x)//求排名 
{
    
    
	if (x==0) return 0;//未查询到结果
	if (v==a[x].v) return a[a[x].l].size;//返回排名
	if (v<a[x].v) return GRBV(a[x].l); 
	return GRBV(a[x].r)+a[a[x].l].size+a[x].cnt;//类似二分 
}
long long GVBR(long long x)//求值 
{
    
    
	if (x==0) return 1e9;//未查询到结果
	if (a[a[x].l].size>=rank1) return GVBR(a[x].l);//二分 
	if (a[a[x].l].size+a[x].cnt>=rank1) return a[x].v;//返回结果 
	rank1-=a[a[x].l].size+a[x].cnt;
	return GVBR(a[x].r);//二分 
}
void zig(long long &x)//右旋(左节点变父节点) 
{
    
    
	long long y=a[x].l;
	a[x].l=a[y].r,a[y].r=x,x=y;
	Up(a[x].r),Up(x);
	return;
}
void zag(long long &x)//左旋(右节点变父节点) 
{
    
    
	long long y=a[x].r;
	a[x].r=a[y].l,a[y].l=x,x=y;
	Up(a[x].l),Up(x);
	return;
}
void Insert(long long &x)//插入 
{
    
    
	if (x==0)//新建节点 
	{
    
    
		x=New();
		return;
	}
	if (v==a[x].v)//已有节点,个数+1 
	{
    
    
		a[x].cnt++,Up(x);
		return;
	}
	if (v<a[x].v)//二分 
	{
    
    
		Insert(a[x].l);
		if (a[x].d<a[a[x].l].d) zig(x);
		Up(x);
		return;
	}
	Insert(a[x].r);
	if (a[x].d<a[a[x].r].d) zag(x);
	Up(x);
	return;
}
long long GP()//求前缀 
{
    
    
	long long ans,x=root;
	while (x)
	{
    
    
		if (v>a[x].v)
		{
    
    
			ans=a[x].v,x=a[x].r;
		}
		else x=a[x].l;
	}
	return ans;
}
long long GN()//求后缀 
{
    
    
	long long ans,x=root;
	while (x)
	{
    
    
		if (v<a[x].v)
		{
    
    
			ans=a[x].v,x=a[x].l;
		}
		else x=a[x].r;
	}
	return ans;
}
void Erase(long long &x)//删除 
{
    
    
	if (x==0) return;
	if (v==a[x].v)
	{
    
    
		if (a[x].cnt>1)
		{
    
    
			a[x].cnt--,Up(x);
			return;//多个,删一个 
		}
		if (a[x].l||a[x].r)//有子节点
		{
    
    
			if (a[x].r==0||a[a[x].l].d>a[a[x].r].d) zig(x),Erase(a[x].r);
			else zag(x),Erase(a[x].l);//旋转,然后删除节点(保证平衡) 
			Up(x);
		}
		else x=0;
		return;
	}
	if (v<a[x].v)
	{
    
    
		Erase(a[x].l);
	}
	else
	{
    
    
		Erase(a[x].r);
	}//二分 
	Up(x);
	return;
}
int main()
{
    
    
	//srand(114514);
	Build();
	long long n2;
	cin>>n2>>n;
	for (long long i=1;i<=n2;i++)
	{
    
    
		cin>>v;
		Insert(root);
	}
	long long o,x,last=0,ans=0;
	while (n--)
	{
    
    
		cin>>o>>x;
		x^=last;
		if (o==1)
		{
    
    
			v=x;
			Insert(root);
		}
		if (o==2)
		{
    
    
			v=x;
			Erase(root);
		}
		if (o==3)
		{
    
    
			v=x;
			last=GRBV(root);
			ans^=last;
		}
		if (o==4)
		{
    
    
			rank1=x+1;
			last=GVBR(root);
			ans^=last;
		}
		if (o==5)
		{
    
    
			v=x;
			last=GP();
			ans^=last;
		}
		if (o==6)
		{
    
    
			v=x;
			last=GN();
			ans^=last;
		}
	}
	cout<<ans;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_49843717/article/details/114645127