来自刚刚学会Treap的蒟蒻

题目链接:

特别不习惯用一堆数组,结构体多香啊

struct node{
	int son[2];//son[
	int size,cnt;
	int val,rd;
}trp[1500000];
int tot,root;

其实完整代码里都有~
只不过无聊,单独拿出来~

int rrand(void){
    return seed=(int)((long long)(seed)*974485642LL%21474836LL);
    /*手写rand()取随机数*/
}
#include<bits/stdc++.h>
#define re register
#define rd(x) x=read()
struct node{
	int son[2];//son[0]左儿子,son[1]右儿子 
	int size,cnt;
	//因为treap不希望出现值相同的节点,(想想为什么) 
	//所以用cnt统计相同的值 
	int val,rd;
	//rd 是随机的权值,整个树按照这个打乱 
	//当然,rd也可以被称为该节点的优先级 
}trp[1500000];
int tot,root;
int seed=5201314;
int rrand(void){
    return seed=(int)((long long)(seed)*974485642LL%21474836LL);
    /*手写rand()取随机数*/
}

void newnode(int &now,int val){
	/* 
	val = val
	size = 1
	cnt = 1;
	rd rand()
	*/
	/*基础的新建节点,没毛病*/ 
	now = ++tot;
	trp[now].val = val;
	trp[now].size = trp[now].cnt = 1;
	trp[now].rd = rrand();
	/*这里吐槽下,treap真的是看脸的算法啊*/
}
void update(int now){
	/*更新新节点,很简单,这里没啥问题 */
	trp[now].size = trp[trp[now].son[0]].size+trp[trp[now].son[1]].size+trp[now].cnt;
}
void rotate(int &now,int d){
	/*
	自闭了十分钟的旋转操作,
	位运算理解成左右儿子切换或者左右方向切换就好。 
	*/
	int son = trp[now].son[d^1];
	trp[now].son[d^1] = trp[son].son[d];
	trp[son].son[d] = now;
	now = son;
	/*注意,从下到上更新,先更新旋转后节点的d儿子,最后更新旋转的now节点*/ 
	update(trp[now].son[d]);
	update(now);
}
void insert(int &now,int val){
	if(!now){
		newnode(now,val);//终于可以插入新节点了 
		return;
	}//在那之前,我们需要找到在何处插
	++trp[now].size;
	if(trp[now].val==val){
		trp[now].cnt++;//前面提到的计数器,相同的值直接统计数量就好
		update(now);//我还在思考这里要不要update一下 
		return;
	}
	if(trp[now].val>val){
		insert(trp[now].son[0],val);
		if(trp[now].rd<trp[trp[now].son[0]].rd)
			rotate(now,1);
	}//左右分头,记住一点,这棵树中,越往上rd越大,所以我们if比较里的内容其实很简单
	//如果遇到儿子的rd比父辈大的时候,我们就旋转一下 
	else{
		insert(trp[now].son[1],val);
		if(trp[now].rd<trp[trp[now].son[1]].rd)
			rotate(now,0);
	}
	update(now);
}
void del(int &now,int val){
	if(!now)return;//都没节点了,你删个锤子,可以加返回值判断是否删成功 
	++trp[now].size;
	if(trp[now].val==val){
		if(trp[now].cnt>1){
			trp[now].cnt--;
			update(now);
			return;
		}
		if(trp[now].son[0]||trp[now].son[1]){
			if(!trp[now].son[1]||trp[trp[now].son[0]].rd>trp[trp[now].son[1]].rd)
				rotate(now,1),del(trp[now].son[1],val);
			else
				rotate(now,0),del(trp[now].son[0],val);
		}//删除这段其实用手画三四张模拟就懂了 
		else now = 0;
	}//蒟蒻的我可能会在B站录个视频什么的 
	else if(trp[now].val>val) 
		del(trp[now].son[0],val);
	else del(trp[now].son[1],val);
	update(now);
}
int find_rank(int now,int val){
	if(!now)
		return INT_MIN;
	if(trp[now].val==val)
		return trp[trp[now].son[0]].size+1;
	else if(trp[now].val>val)
		return find_rank(trp[now].son[0],val);
	else
		return trp[trp[now].son[0]].size + trp[now].cnt + find_rank(trp[now].son[1],val);
}
int find_val(int now,int rank){
	if(!now)
		return INT_MAX;
	if(trp[trp[now].son[0]].size>=rank)
		return find_val(trp[now].son[0],rank);
	else if(trp[trp[now].son[0]].size+trp[now].cnt>=rank)
		return trp[now].val;
	else
		return find_val(trp[now].son[1],rank-trp[trp[now].son[0]].size-trp[now].cnt);
}
int find_pre(int val){
	re int now = root,pre = 0;
	while(now){
		if(trp[now].val<val)
			pre = trp[now].val,now = trp[now].son[1];
		else
			now = trp[now].son[0];
	}
	return pre;
}
int find_nex(int val){
	re int now = root,nex = 0;
	while(now){
		if(trp[now].val>val)
			nex = trp[now].val,now = trp[now].son[0];
		else
			now = trp[now].son[1];
	}
	return nex;
}
inline int read(){
	re int ans = 0;
	re bool f = 1;
	re char ch = getchar();
	while(ch<'0'||ch>'9')ch=='-'?f=0,ch=getchar():ch=getchar();
	while(ch>='0'&&ch<='9')ans =(ans<<3)+(ans<<1)+(ch^48),ch = getchar(); 
	return f?ans:-ans;
} 
signed main(void){
	int n,m,last = 0,ans = 0,ooo;
	n = read();m = read();++n;++m;
	while(--n){
		ooo = read();
		insert(root,ooo);
	}
	while(--m){
		int opt,x;
		opt = read();
		x = read();
		switch(opt){
			case 1:
				insert(root,x^last);
				break;
			case 2:
				del(root,x^last);
				break;
			case 3:
				ans ^= (last = find_rank(root,x^last));
				break;
			case 4:
				ans ^= (last = find_val(root,x^last));
				break;
			case 5:
				ans ^= (last = find_pre(x^last));
				break;
			case 6:
				ans ^= (last = find_nex(x^last));
				break;			
		}
	}
	printf("%d",ans);
	return 0;
}

感谢观看,如有不懂欢迎评论,博主全年365天无休,24小时内必回

嘿嘿,点个关注推荐呗~

版权声明:本文为Histone原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://www.cnblogs.com/Histone/p/12609280.html