c++使用树状数组维护的普通平衡树（可以过洛谷题）

函数：add,sum,query,two_point;

通过这四种函数的组合求前驱，后继;

add（正常的树状数组函数），code:(没什么可解释的）

void add(int x,int v){
	for(int i=x;i<=tot;i+=lowbit(i)){
		tr[i]+=v;
	}
}

two_point，求某个数在q数组（存储数值的数组）的位置，可以使用lower_bound和upper_bound来代替：code:

int two_point(int x){
	int l=1,r=tot;
	while(l<r){
		int mid=l+r>>1;
		if(q[mid]>=x) r=mid;
		else l=mid+1;
	}
	return r;
}

sum（正常的树状数组函数），code:(没什么可解释的）

int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i)){
		res+=tr[i];
	}
	return res;
}

query(通过排名查询数本身），code{

int query(int x){
	int t=0;
	for(int i=19;i>=0;i--){
		int net=t+(1<<i);//查找t的父节点
		if(net<=tot&&tr[net]<x){//要求t合法，x为查找的数
			t=net;//t变成他的父节点
			x-=tr[t];//x逐渐减少，树状数组常规操作
		}
	}
	return q[t+1];//因为下标从一开始
}

6种操作对应函数调用：

        if(p[i]==1) add(two_point(a[i]),1);//插入数值 x。
		if(p[i]==2) add(two_point(a[i]),-1);//删除数值 x(若有多个相同的数，应只删除一个)。
		if(p[i]==3) cout<<sum(two_point(a[i])-1)+1<<endl;//查询数值 x 的排名(若有多个相同的数，应输出最小的排名)。
		if(p[i]==4) cout<<query(a[i])<<endl;//查询排名为 x 的数值。
		if(p[i]==5)	cout<<query(sum(two_point(a[i])-1))<<endl;//求数值 x 的前驱(前驱定义为小于 x 的最大的数)
		if(p[i]==6) cout<<query(sum(two_point(a[i]))+1)<<endl;//求数值 x 的后继(后继定义为大于 x 的最小的数)。

总：

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int tr[N],q[N],a[N],p[N],tot=0;
int lowbit(int x){
	return x&-x;
}
int two_point(int x){
	int l=1,r=tot;
	while(l<r){
		int mid=l+r>>1;
		if(q[mid]>=x) r=mid;
		else l=mid+1;
	}
	return r;
}
void add(int x,int v){
	for(int i=x;i<=tot;i+=lowbit(i)){
		tr[i]+=v;
	}
}
int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i)){
		res+=tr[i];
	}
	return res;
}
int query(int x){
	int t=0;
	for(int i=19;i>=0;i--){
		int net=t+(1<<i);
		if(net<=tot&&tr[net]<x){
			t=net;
			x-=tr[t];
		}
	}
	return q[t+1];
}
int main(){
	int n;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>p[i]>>a[i];
		if(p[i]!=4) q[++tot]=a[i];
	}
	sort(q+1,q+1+tot);
	tot=unique(q+1,q+tot+1)-(q+1);
	for(int i=0;i<n;i++){
		if(p[i]==1) add(two_point(a[i]),1);
		if(p[i]==2) add(two_point(a[i]),-1);
		if(p[i]==3) cout<<sum(two_point(a[i])-1)+1<<endl;
		if(p[i]==4) cout<<query(a[i])<<endl;
		if(p[i]==5)	cout<<query(sum(two_point(a[i])-1))<<endl;
		if(p[i]==6) cout<<query(sum(two_point(a[i]))+1)<<endl;
	}
	return 0;
}

喜欢就点个赞吧！