普通平衡树

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<string>
#include<vector>
#include<string.h>
#include<map>
#include<cmath>
#include<set>
#define ll long long
#define maxn 1000001
#define ull unsigned long long
using namespace std;
#define N 100005
#define M 10000005
#define mod 100000000

struct node{
	int opt,x;
}a[N];

int n,ans,f1[M*2+100],f2[N],b[N],c[N];//f1[M<<1+100]不行

struct Node{
	int l,r,w;
}tree[N<<2];

void build(int l,int r,int k){//建树
	tree[k].l=l,tree[k].r=r;
	if(l==r) return;
	int mid=(l+r)>>1;
	build(l,mid,k<<1);
	build(mid+1,r,k<<1|1);
}
void update(int x,int w,int k){//在x个位置插入或删除数
	if(tree[k].l==tree[k].r){
		tree[k].w+=w;//这个区间数加一或减一
		return;
	}
	int mid=(tree[k].l+tree[k].r)>>1;
	if(x<=mid) update(x,w,k<<1);
	else update(x,w,k<<1|1);
	tree[k].w=tree[k<<1].w+tree[k<<1|1].w;//这个区间几个数
}

void Xth(int x,int k){//第x小值
	if(tree[k].l==tree[k].r){
		ans=tree[k].l;
		return;
	}
	if(tree[k<<1].w>=x) Xth(x,k<<1);
	else Xth(x-tree[k<<1].w,k<<1|1);//是 “第”
}

void Rank(int x,int k){//x的排名,就是求1到x-1的区间和
	if(tree[k].l>=1&&tree[k].r<=x){
		ans+=tree[k].w;
		return;
	}
	int mid=(tree[k].l+tree[k].r)>>1;
	if(1<=mid) Rank(x,k<<1);//注意
	if(x>mid) Rank(x,k<<1|1);
}
int Findr(int k){//找到这结点k区间的最右边的数
	if(tree[k].l==tree[k].r) return tree[k].l;
	if(tree[k<<1|1].w) return Findr(k<<1|1);
	return Findr(k<<1);
}
int Findl(int k){//找到这结点k区间的最左边的数
	if(tree[k].l==tree[k].r) return tree[k].l;
	if(tree[k<<1].w) return Findl(k<<1);
	return Findl(k<<1|1);
}
int Pre(int x,int k){//x的前驱
	if(tree[k].r<x){
		if(tree[k].w) return Findr(k);
		return 0;
	}
	int mid=(tree[k].l+tree[k].r)>>1,res=0;
	if(mid<x-1&&tree[k<<1|1].w&&(res=Pre(x,k<<1|1))) return res;
	return Pre(x,k<<1);
}

int Nex(int x,int k){//x的后驱
	if(tree[k].l>x){
		if(tree[k].w) return Findl(k);
		return 0;
	}
	int mid=(tree[k].l+tree[k].r)>>1;
	int res=0;
	if(x<mid&&tree[k<<1].w&&(res=Nex(x,k<<1))) return res;
	return Nex(x,k<<1|1);
}

int main(){
    ios::sync_with_stdio(false);
    cin>>n;
    int cnt=0;
    for(int i=1;i<=n;i++){
    	cin>>a[i].opt>>a[i].x;
    	if(a[i].opt==1||a[i].opt==3||a[i].opt==5||a[i].opt==6){
    		cnt++;
    		b[cnt]=a[i].x;
    		c[cnt]=b[cnt];//b数组要动,存个c数组
		}
	}
	sort(b+1,b+1+cnt);//排序
	int t=cnt,pon;//cnt要动,存个t
	cnt=unique(b+1,b+cnt+1)-b-1;//去重后有几个元素
	for(int i=1;i<=t;i++){//离散化,所有的数
		pon=lower_bound(b+1,b+cnt+1,c[i])-b;//在去重后的数组里找大于等于c[i]的是 第几个元素
		f1[c[i]+M]=pon;//互相对应记录
		f2[pon]=c[i];//pon排名对应的数c[i]
	}
	build(1,N,1);
	for(int i=1;i<=n;i++){
		if(a[i].opt==1) update(f1[a[i].x+M],1,1);//加
		else if(a[i].opt==2) update(f1[a[i].x+M],-1,1);//减
		else if(a[i].opt==3){
			ans=0;
			Rank(f1[a[i].x+M]-1,1);
			cout<<ans+1<<endl;
		}
		else if(a[i].opt==4){
			ans=0;
			Xth(a[i].x,1);
			cout<<f2[ans]<<endl;
		}
		else if(a[i].opt==5){
			cout<<f2[Pre(f1[a[i].x+M],1)]<<endl;
		}
		else{
			cout<<f2[Nex(f1[a[i].x+M],1)]<<endl;
		}
	}
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

When_3*four

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值