// 树状数组的写法（插入，删除，查询，前驱，后继） P3369 【模板】普通平衡树

luogu

树状数组的板子

题目描述

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：

    插入xxx数
    删除xxx数(若有多个相同的数，因只删除一个)
    查询xxx数的排名(排名定义为比当前数小的数的个数+1+1+1。若有多个相同的数，因输出最小的排名)
    查询排名为xxx的数
    求xxx的前驱(前驱定义为小于xxx，且最大的数)
    求xxx的后继(后继定义为大于xxx，且最小的数)

输入格式

第一行为nnn，表示操作的个数,下面nnn行每行有两个数optoptopt和xxx，optoptopt表示操作的序号( 1≤opt≤6 1 \leq opt \leq 6 1≤opt≤6 )
输出格式

对于操作3,4,5,63,4,5,63,4,5,6每行输出一个数，表示对应答案
输入输出样例
输入 #1

10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598

输出 #1

106465
84185
492737

说明/提示

时空限制：1000ms,128M

1.n的数据范围： n≤100000 n \leq 100000 n≤100000

2.每个数的数据范围： [−107,107][-{10}^7, {10}^7][−107,107]

来源：Tyvj1728 原名：普通平衡树

在此鸣谢

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100010;
ll c[N],p[N],num[N],nn,n;
int bz[N];
int lowbit(int x)
{
	return x&(-x);
}
void add(int x,ll v)
{
	while(x<=nn)
	{
		c[x]+=v;
		x+=lowbit(x);
	}
}
int P(ll x)
{
	return lower_bound(p+1,p+1+nn,x)-p;
}
ll sum(int x)
{
	ll ans=0;
	while(x)
	{
		ans+=c[x];
		x-=lowbit(x);
	}
	return ans;
}
int find_kk(int x)
{
	ll ans=0,cnt=0;
	for(int i=20;i>=0;i--)
	{
		ans+=(1<<i);
		if(ans>=nn||cnt+c[ans]>=x)
		{
			ans-=(1<<i);
		}else cnt+=c[ans];
	}
	return ans+1;
}
int main()
{
	
	scanf("%d",&n);int tot=0;
	for(int i=1;i<=n;i++)
	{
		scanf("%d%lld",&bz[i],&num[i]);
		if(bz[i]!=4) p[++tot]=num[i];
	}
	sort(p+1,p+1+tot);
	int pos=unique(p+1,p+1+tot)-p-1; nn=pos;
	for(int i=1;i<=n;i++)
	{
		ll x=num[i];
		if(bz[i]==1) add(P(x ),1);
		if(bz[i]==2) add(P(x ),-1);
		if(bz[i]==3) printf("%lld\n",sum(P(x)-1)+1);//x数的排名 
		if(bz[i]==4) printf("%lld\n",p[find_kk(x)]);//排名为x的数 
		if(bz[i]==5) printf("%lld\n",p[find_kk(sum(P(x)-1))]);
		if(bz[i]==6) printf("%lld\n",p[find_kk(sum(P(x))+1)]);
	}
}

或者

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100010;
ll c[N],p[N],num[N],nn,n;
int bz[N];
int lowbit(int x)
{
	return x&(-x);
}
void add(int x,ll v)
{
	while(x<=nn)
	{
		c[x]+=v;
		x+=lowbit(x);
	}
}
int P(ll x)
{
	return lower_bound(p+1,p+1+nn,x)-p;
}
ll sum(int x)
{
	ll ans=0;
	while(x)
	{
		ans+=c[x];
		x-=lowbit(x);
	}
	return ans;
}
int find_kk(int x)
{
	ll ans=0,cnt=0;
	for(int i=20;i>=0;i--)
	{
		ans+=(1<<i);
		if(ans>nn||cnt+c[ans]>=x)
		{
			ans-=(1<<i);
		}else cnt+=c[ans];
	}
	return ans+1;
}
int main()
{
	
	scanf("%d",&n);int tot=0;
	for(int i=1;i<=n;i++)
	{
		scanf("%d%lld",&bz[i],&num[i]);
		p[++tot]=num[i];
	}
	sort(p+1,p+1+tot);
	int pos=unique(p+1,p+1+tot)-p-1; nn=pos;
	for(int i=1;i<=n;i++)
	{
		ll x=num[i];
		if(bz[i]==1) add(P(x ),1);
		if(bz[i]==2) add(P(x ),-1);
		if(bz[i]==3) printf("%lld\n",sum(P(x)-1)+1);//x数的排名 
		if(bz[i]==4) printf("%lld\n",p[find_kk(x)]);//排名为x的数 
		if(bz[i]==5) printf("%lld\n",p[find_kk(sum(P(x)-1))]);//x的最大前驱
		if(bz[i]==6) printf("%lld\n",p[find_kk(sum(P(x))+1)]);//x的最小后继
	}
}

参考