28分 初学Treap 求大佬帮忙调调板子题 QAQ
查看原帖
28分 初学Treap 求大佬帮忙调调板子题 QAQ
335552
Christophe_楼主2022/3/8 22:21
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<stdlib.h>
using namespace std;
const int N=1e5+6,INF=1e7+1e6+6;
int n,root,tot;
struct Treap{
	int lc,rc;//左、右儿子
	int val,dat;//val:真实值, dat:辅助堆的权值
	int cnt,size;//cnt:本身出现的次数,size:子树cnt+本身cnt
}tr[N];
inline int New(int val){
	tr[++tot]={0,0,val,rand(),1,1};
	return tot;
}
inline void Update(int p){ 
	tr[p].size=tr[tr[p].lc].size+tr[tr[p].rc].size+tr[p].cnt; 
}
inline void Build(){ 
	root=New(-INF),tr[root].rc=New(INF),Update(root); 
}
inline void zig(int &p){
	int q=tr[p].lc;
	tr[p].lc=tr[q].rc,tr[q].rc=p,p=q;
	Update(tr[p].rc),Update(p);//更新时要从下向上
}
inline void zag(int &p){
	int q=tr[p].rc;
	tr[p].rc=tr[q].lc,tr[q].lc=p,p=q;
	Update(tr[p].lc),Update(p);//更新时要从下向上	
}
void Insert(int &p,int val){
	if(p==0){ 
		p=New(val);
		return;
	}
    if(val==tr[p].val){
    	++tr[p].cnt,Update(p);
    	return;
	}
	if(val<tr[p].val){
		Insert(tr[p].lc,val);
		if(tr[tr[p].lc].dat>tr[p].dat) zig(p);
	}else{
		Insert(tr[p].rc,val);
		if(tr[tr[p].rc].dat>tr[p].dat) zag(p);		
	}
	Update(p);
}
void Remove(int &p,int val){
	if(p==0) return;
	if(val==tr[p].val){
	    if(tr[p].cnt>1){
	        --tr[p].cnt,Update(p);
	        return;
	    }
		if(tr[p].lc||tr[p].rc){
			if(tr[p].rc==0||tr[tr[p].lc].dat>tr[tr[p].rc].dat) zig(p),Remove(tr[p].lc,val);
			else zag(p),Remove(tr[p].rc,val);
			Update(p);
		}else p=0;
		return;
	}
	if(val<tr[p].val) Remove(tr[p].lc,val);
	else Remove(tr[p].rc,val);
	Update(p);
}  
int GetRankByVal(int p,int val){
	if(p==0) return 0;
	if(val==tr[p].val) return tr[tr[p].lc].size+1;//注意是左孩子+1,而不是本身
	if(val<tr[p].val) return GetRankByVal(tr[p].lc,val);
	return tr[tr[p].lc].size+GetRankByVal(tr[p].rc,val)+tr[p].cnt;//注意是tr[p].cnt而不是1(因为tr[p].val<val而不等于) 
}
int GetValByRank(int p,int rank){
	if(p==0) return INF;
	if(rank<=tr[tr[p].lc].size) return GetValByRank(tr[p].lc,rank);//注意是 <= 左子树 而非 < ;
	if(rank<=tr[tr[p].lc].size+tr[p].cnt) return tr[p].val;
    return GetValByRank(tr[p].rc,rank-tr[tr[p].lc].size-tr[p].cnt);
}
int GetPre(int p,int val){
	int ans=1;
	while(p){
		if(val==tr[p].val){
			if(tr[p].lc){
				int q=tr[p].lc;
				while(tr[q].rc) q=tr[q].rc;
				ans=q;
			}
			break;
		}
		if(tr[p].val<val&&tr[p].val>tr[ans].val) ans=p;
		if(val<tr[p].val) p=tr[p].lc;
		else p=tr[p].rc; 
	}
	return tr[ans].val;
}
int GetNext(int p,int val){
	int ans=2;
	while(p){
		if(val==tr[p].val){
			if(tr[p].rc){
				int q=tr[p].rc;
				while(tr[q].lc) q=tr[q].lc;
				ans=q;
			}
			break;
		}
		if(tr[p].val>val&&tr[p].val<tr[ans].val) ans=p;
		if(val<tr[p].val) p=tr[p].lc;
		else p=tr[p].rc; 
	}
	return tr[ans].val;
}
int main(){
	Build();
	scanf("%d",&n);
	for(int i=1;i<=n;++i){
		int op,x;
		scanf("%d%d",&op,&x);
		if(op==1) Insert(root,x);
		else if(op==2) Remove(root,x);
		else if(op==3) printf("%d\n",GetRankByVal(root,x)-1);
		else if(op==4) printf("%d\n",GetValByRank(root,x+1));	
		else if(op==5) printf("%d\n",GetPre(root,x));	
		else if(op==6) printf("%d\n",GetNext(root,x));	
	}
	return 0;
}

过了样例和测试点 1、2、12 ,其他是一片喜庆的大红色 qwq .

2022/3/8 22:21
加载中...