蒟蒻求调
查看原帖
蒟蒻求调
930914
Pumpkin_vs_Earth楼主2024/1/17 11:06
题目传送门
#include<bits/stdc++.h>
#define maxm 10000005
using namespace std;
int cnt;
struct node{
	int ls,rs;
	int key,pri;
	int size;
}t[maxm*4];
void newnode(int x){
	cnt++;
	t[cnt].size=1;
	t[cnt].ls=t[cnt].rs=0;
	t[cnt].key=x;
	t[cnt].pri=rand();
}
void update(int u){
	t[u].size=t[t[u].ls].size+t[t[u].rs].size+1;
}
void rotate(int &o,int d){
	int k;
	if(d==1){
		k=t[o].rs;
		t[o].rs=t[k].ls;
		t[k].ls=o;
	}
	else{
		k=t[o].ls;
		t[o].ls=t[k].rs;
		t[k].rs=o;
	}
	t[k].size=t[o].size;
	update(o);
	o=k;
}
void insert(int &u,int x){
	if(u==0){
		newnode(x);
		u=cnt;
		return;
	}
	t[u].size++;
	if(x>=t[u].key)insert(t[u].rs,x);
	else insert(t[u].ls,x);
	if(t[u].ls!=0&&t[u].pri>t[t[u].ls].pri)rotate(u,0);
	if(t[u].rs!=0&&t[u].pri>t[t[u].rs].pri)rotate(u,1);
	update(u);
}
void del(int &u,int x){
	t[u].size--;
	if(t[u].key==x){
		if(t[u].ls==0&&t[u].rs==0){
			u=0;
			return;
		}
		if(t[u].ls==0||t[u].rs==0){
			u=t[u].ls+t[u].rs;
			return;
		}
		if(t[t[u].ls].pri<t[t[u].rs].pri){
			rotate(u,0);
			del(t[u].rs,x);
			return;
		}
		else{
			rotate(u,1);
			del(t[u].ls,x);
			return;
		}
	}
	if(t[u].key>=x)del(t[u].ls,x);
	else del(t[u].rs,x);
	update(u);
}
int Rank(int u,int x){
	if(u==0)return 0;
	if(x>t[u].key)return t[t[u].ls].size+1+Rank(t[u].rs,x);
	return Rank(t[u].ls,x);
}
int kth(int u,int k){
	if(k==t[t[u].ls].size+1)return t[u].key;
	if(k>t[t[u].ls].size+1)return kth(t[u].rs,k-t[t[u].ls].size-1);
	if(k<=t[t[u].ls].size)kth(t[u].ls,k);
}
int precu(int u,int x){
	if(u==0)return 0;
	if(t[u].key>=x)return precu(t[u].ls,x);
	int tmp=precu(t[u].rs,x);
	if(tmp==0)return t[u].key;
	return tmp;
}
int succ(int u,int x){
	if(u==0)return 0;
	if(t[u].key<=x)return succ(t[u].rs,x);
	int tmp=succ(t[u].ls,x);
	if(tmp==0)return t[u].key;
	return tmp;
}
signed main(){
//	ios::sync_with_stdio(0);
//	cin.tie(0);
//	cout.tie(0);
	srand(time(NULL));
	int root=0;
	int n;
	cin>>n;
	while(n--){
		int opt,x;
		cin>>opt>>x;
		switch(opt){
			case 1: insert(root,x);break;
			case 2: del(root,x);break;
			case 3: cout<<Rank(root,x)+1<<endl;break;
			case 4: cout<<kth(root,x)<<endl;break;
			case 5: cout<<precu(root,x)<<endl;break;
			case 6: cout<<succ(root,x)<<endl;break;
		}
	}
	return 0;
} 

测试点情况

2024/1/17 11:06
加载中...