洛谷 P2617 Dynamic Rankings 树状数组套主席树 动态求区间第k大

P2317

思路以及代码是学习这位大佬的： GuessYCB

建好主席树后，每当修改某个数时，还要修改其后续的线段树，可以借用树状数组，存的是每颗线段树的第一个节点，用树状数组去更新后续的线段树，复杂度logn*logn，那么如何去查询呢，将主席树的查询以及树状数组的查询结合就好了，比如查找1--n的第k大，先用树状数组快速找到需要查询的线段树，假设找到了x个线段树，接下来就是主席树的查询方法了，将这x个线段树的左子树的sum值累加与k比较即可，如果k<=sum，接下来就查询这x个线段树的左子树，否则查询这x个线段树的右子树。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=2e6+5;
int rt[maxn*2],ls[maxn*20],rs[maxn*20],sum[maxn*20];
int n,tot,sz,cnt,a[maxn],b[maxn*2],c[maxn];
int V1[maxn],V2[maxn],c1,c2;
struct node
{
	int l,r,k,flag;
}q[maxn];
void update(int &o,int l,int r,int k,int value)
{
	if(!o)o=++cnt;
	sum[o]+=value;
	if(l==r)return;
	int m=(l+r)/2;
	if(k<=m)update(ls[o],l,m,k,value);
	else update(rs[o],m+1,r,k,value); 
}
void modify(int pos,int value)
{
	int k=lower_bound(b+1,b+1+sz,a[pos])-b;
	for(int i=pos;i<=n;i+=(i&-i))
	update(rt[i],1,sz,k,value);
}
int query(int l,int r,int k)
{
	if(l==r)return l;
	int m=(l+r)/2,res=0;
	for(int i=1;i<=c1;i++)res+=sum[ls[V1[i]]];
	for(int i=1;i<=c2;i++)res-=sum[ls[V2[i]]];
	if(k<=res)
	{
		for(int i=1;i<=c1;i++)V1[i]=ls[V1[i]];
		for(int i=1;i<=c2;i++)V2[i]=ls[V2[i]];
		return query(l,m,k);
	}
	else
	{
		for(int i=1;i<=c1;i++)V1[i]=rs[V1[i]];
		for(int i=1;i<=c2;i++)V2[i]=rs[V2[i]];
		return query(m+1,r,k-res);		
	}
}
int get(int l,int r,int k)
{
	c1=c2=0;
	for(int i=r;i;i-=(i&-i))
	V1[++c1]=rt[i];
	for(int i=l-1;i;i-=(i&-i))
	V2[++c2]=rt[i];
	return query(1,sz,k);
}
int main()
{
	int m,l,r,k;
	char s[2];
	tot=0;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]),b[++tot]=a[i];
	for(int i=1;i<=m;i++)
	{
		scanf("%s",s);
		if(s[0]=='C')
		{
			scanf("%d%d",&q[i].l,&q[i].k);
			q[i].flag=1,b[++tot]=q[i].k;
		}
		else
		{
			scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].k);
			q[i].flag=2;
		}		
	}
	sort(b+1,b+1+tot);
	sz=unique(b+1,b+1+tot)-(b+1);
	for(int i=1;i<=n;i++)
	modify(i,1);
	for(int i=1;i<=m;i++)
	{
		if(q[i].flag==1)
		{
			modify(q[i].l,-1);
			a[q[i].l]=q[i].k;
			modify(q[i].l,1);
		}
		else
		printf("%d\n",b[get(q[i].l,q[i].r,q[i].k)]);
	}
}