非旋treap(fhq treap) 指针版

传送门
看了一圈,好像真的没什么用指针的呢。。

明明觉得指针很好看(什么??你说RE???听不见听不见)
其实我觉得用数组的话不RE直接WA调起来不是更困难嘛,毕竟通过gdb还可以知道哪里RE,WA就不知道咋回事了,是不是很有道理,虽然我还是调了几小时

我写的是fhq treap,核心是split和merge操作,思想高赞dalao都讲得很清楚,我语文弱渣就不班门弄斧了,主要是想提供一个指针版的参考吧QAQ

我真的是一整天都在搞分裂(split),有种要进入七月枪毙名单的赶脚,慌张.jpg

#include<bits/stdc++.h>
#define LL long long
#define fr(i,x,y) for(int i=(x);i<=(y);i++)
#define rf(i,x,y) for(int i=(x);i>=(y);i--)
#define frl(i,x,y) for(int i=(x);i<(y);i++)
using namespace std;
const int N=100005;
struct node{
	int v,rnd,s,sz;  //s表示权值为v的个数,sz表示子树size,然而我经常忘记s的存在,直接写成1,挂了好久
	node* ch[2];
	
	inline int cmp(int x){ return x>v; } //这其实是写旋转treap时留下的历史遗留问题= =无视吧
	
	inline void maintain(){
		sz=s;
		if (ch[0]!=NULL) sz+=ch[0]->sz; //写指针一定要特别注意对NULL的判断
		if (ch[1]!=NULL) sz+=ch[1]->sz;
	}
	
	node(){
		ch[0]=ch[1]=NULL;
		rnd=rand();
	}
}nd[N];
int tot;
node* rt;
int n;

void read(int &x){ //读优一开始忘记负数了= =
	char ch=getchar();x=0;int w=0;
	for(;ch<'0'||ch>'9';ch=getchar()) if (ch=='-') w=1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=(x<<3)+(x<<1)+ch-'0';
	if (w) x=-x;
}

inline int sss(node* &o){
	return o==NULL?0:o->sz;
}

node* &kth(node* &o,int k){ 
	assert(o!=NULL);
	int s=sss(o->ch[0]);
	if (s+1<=k&&s+o->s>=k) return o;
	if (s+1>k) return kth(o->ch[0],k);
	return kth(o->ch[1],k-o->s-s);
}

void split(node* o,node* &L,node* &r,int k){ //split <=k
	if (o==NULL) return L=r=NULL,void();
	if (o->v<=k){
		split(o->ch[1],o->ch[1],r,k);
		L=o;
	} else{
		split(o->ch[0],L,o->ch[0],k);
		r=o;
	}
	o->maintain(); //要经常维护一下信息
}

node* &merge(node* &L,node* &r){
	if (L==NULL) return r;
	if (r==NULL) return L;
	if (L->rnd<r->rnd){
		L->ch[1]=merge(L->ch[1],r);
		L->maintain();
		return L;
	} else{
		r->ch[0]=merge(L,r->ch[0]);
		r->maintain();
		return r;
	}
}

void add_node(int v){
	node* L;
	node* r;node* xx;
	split(rt,L,r,v);
	if (L!=NULL&&(xx=kth(L,L->sz))->v==v){ //如果存在这个数直接加个数
		split(L,L,xx,v-1);
		xx->s++;xx->sz++;  //不要忘记加size
		//rt=merge(L,r);
	}else{
		xx=&nd[++tot];
		xx->s=xx->sz=1;xx->v=v;
	}
	rt=merge(merge(L,xx),r);
}

void del_node(int v){
	node *L,*r,*mid;
	split(rt,L,r,v-1);
	split(r,mid,r,v);
	if (mid!=NULL&&mid->s>1) mid->s--,mid->sz--,r=merge(mid,r);
	rt=merge(L,r);
}

int rk(node* &o,int v){
	assert(o!=NULL);
	if (o->v==v) return sss(o->ch[0])+1;
	if (v<o->v) return rk(o->ch[0],v);
	 else return rk(o->ch[1],v)+sss(o->ch[0])+o->s;
}

int rkk(node* rt,int v){
	node *L,*r;
	split(rt,L,r,v-1);
	int ans=sss(L)+1;
	rt=merge(L,r);
	return ans;
}
//rk和rkk都可以求rank,一个通过split一个通过size,好像rkk更快?感觉有点奇怪...

int main(){
	srand(19260817);
	read(n);
	int tp,x;
	node *L,*r;
	int s=0;
	add_node(19260817);
	fr(o,1,n){
		read(tp);read(x);
		if (tp==1) add_node(x);
		if (tp==2) del_node(x);
		if (tp==3) printf("%d\n",rk(rt,x)),s++;
		if (tp==4) printf("%d\n",kth(rt,x)->v),s++;
		if (tp==5){
			split(rt,L,r,x-1);
			printf("%d\n",kth(L,L->sz)->v),s++;
			rt=merge(L,r);
		}
		if (tp==6){
			split(rt,L,r,x);
			printf("%d\n",kth(r,1)->v),s++;
			rt=merge(L,r);
		}
		//if (s==670) printf("----%d %d %d\n",o,tp,x);
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值