动态第k大（树状数组套主席树）

P2617 Dynamic Rankings

这道题要求单点修改+区间查询第k大

因此可以用树状数组维护权值，主席树维护位置

其中主席树的第i个根代表的是$[i-lowbit(i)+1,i]$的信息

查询的时候则是查询$[l-1,r]$的区间，利用前缀和的思想进行查询，右减左

另外，需要注意的是，主席树的建树，因为一个根代表一个区间，而不是一个点，因此，不需要依赖前面的点，直接在这棵树上修改即可，即每一个树状数组都对应一棵独立的主席树。

具体来说就是这样

insert(rt[i],rt[i],1,cnt,k,y);

修改时，树状数组中元素代表区间含有改下标的都要修改

#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
#define inf 0x7fffffff
//#define ll long long
//#define int long long
//#define double long double
#define re int
#define void inline void
#define eps 1e-8
//#define mod 1e9+7
#define ls(p) p<<1
#define rs(p) p<<1|1
#define pi acos(-1.0)
#define pb push_back
#define P pair < int , int > 
using namespace std;
const int p=1e9+7;
const int M=2e3+5;
const int N=5e5+5;//?????????? 4e8
struct node
{
	int l,r,sum;
}e[N*80];
int a[N],b[N],rt[N],tot,n,m,q;
int xx[N],yy[N],tx,ty;
int cnt=0,cl[N],cr[N],ck[N];
int lowbit(int x)
{
	return  x&(-x);
}
void insert(int &p,int pre,int l,int r,int d,int v)
{
	e[++tot]=e[pre];
	p=tot;
	e[tot].sum+=v;
	if(l==r)  return;
	int mid=(l+r)>>1;
	if(d<=mid)  insert(e[p].l,e[pre].l,l,mid,d,v);
	else  insert(e[p].r,e[pre].r,mid+1,r,d,v);
}
int ask(int l,int r,int k)
{
	if(l==r)  return l;
	int mid=(l+r)>>1;
	int x=0;
	for(re i=1;i<=tx;i++)  x-=e[e[xx[i]].l].sum;
	for(re i=1;i<=ty;i++)  x+=e[e[yy[i]].l].sum;
	if(x>=k)
	{
		for(re i=1;i<=tx;i++)  xx[i]=e[xx[i]].l;
		for(re i=1;i<=ty;i++)  yy[i]=e[yy[i]].l;
		return ask(l,mid,k);
	}
	else
	{
		for(re i=1;i<=tx;i++)  xx[i]=e[xx[i]].r;
		for(re i=1;i<=ty;i++)  yy[i]=e[yy[i]].r;
		return ask(mid+1,r,k-x);
	}
}
void add(int x,int y)
{
	int k=lower_bound(b+1,b+cnt+1,a[x])-b;
//	cout<<k<<endl;
	for(re i=x;i<=n;i+=lowbit(i))  insert(rt[i],rt[i],1,cnt,k,y);
}
void solve()
{
	cin>>n>>q;
	for(re i=1;i<=n;i++)  scanf("%d",&a[i]),b[i]=a[i];
	cnt=n;
	for(re i=1;i<=q;i++)
	{
		char op[2];
		int x,y,k;
		scanf("%s%d",op,&x);
		if(op[0]=='Q')
		{
			scanf("%d%d",&y,&k);
			cl[i]=x,cr[i]=y,ck[i]=k;
		}
		else
		{
			scanf("%d",&y);
			cl[i]=x,cr[i]=y;
			b[++cnt]=y;
		}
	}
	sort(b+1,b+cnt+1);
	cnt=unique(b+1,b+cnt+1)-(b+1);
	for(re i=1;i<=n;i++)  add(i,1);
	for(re i=1;i<=q;i++)  if(ck[i])
	{
		tx=ty=0;
		for(re j=cl[i]-1;j;j-=lowbit(j))  xx[++tx]=rt[j];
		for(re j=cr[i];j;j-=lowbit(j))  yy[++ty]=rt[j];	
		printf("%d\n",b[ask(1,cnt,ck[i])]);
	}
	else
	{
		add(cl[i],-1);
		a[cl[i]]=cr[i];
		add(cl[i],1);
	}
	
}
signed main()
{
//	ios::sync_with_stdio(false);
	int T=1;
//	cin>>2T;
	for(re i=1;i<=T;i++)
	{
		solve();
	}
	
	return 0;
}