解析:
把每个主席树当作成树状数组的一个节点,区间更新的时候,利用树状数组更新,然后在更新到每个版本的主席树
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10000;
int root[N*2],t1[N],t2[N],ls[N*400],rs[N*400],idx;
int a[N],b[N*3],num;
int n,m,n1,n2;
int l[N],r[N],d[N];
int val[N*400];
int lowbit(int x)
{
return x&(-x);
}
void Add(int &rt,int l,int r,int x,int c)
{
if(!rt) rt=++idx;
val[rt]+=c;
if(l==r) return ;
int mid=l+r>>1;
if(x<=mid) Add(ls[rt],l,mid,x,c);
else Add(rs[rt],mid+1,r,x,c);
}
void Change(int pos,int val)
{
int x=lower_bound(b+1,b+1+num,a[pos])-b;
for(int i=pos;i<=n;i+=lowbit(i))
Add(root[i],1,num,x,val);
}
int Kth(int l,int r,int k)
{
if(l==r) return r;
int mid=l+r>>1,sum=0;
for(int i=1;i<=n2;i++) sum+=val[ls[t2[i]]];
for(int i=1;i<=n1;i++) sum-=val[ls[t1[i]]];
if(sum>=k)
{
for(int i=1;i<=n2;i++) t2[i]=ls[t2[i]];
for(int i=1;i<=n1;i++) t1[i]=ls[t1[i]];
return Kth(l,mid,k);
}
else
{
for(int i=1;i<=n2;i++) t2[i]=rs[t2[i]];
for(int i=1;i<=n1;i++) t1[i]=rs[t1[i]];
return Kth(mid+1,r,k-sum);
}
}
int Kth_pre(int l,int r,int k)
{
n1=n2=0;
for(int i=l-1;i>=1;i-=lowbit(i)) t1[++n1]=root[i];
for(int i=r;i>=1;i-=lowbit(i)) t2[++n2]=root[i];
return Kth(1,num,k);
}
int main()
{
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[++num]=a[i];
for(int i=1;i<=m;i++)
{
char ch[1000];
scanf("%s",ch);
if(*ch=='Q')
{
scanf("%d %d %d",&l[i],&r[i],&d[i]);
}
else
{
scanf("%d %d",&l[i],&d[i]);
b[++num]=d[i];
}
}
sort(b+1,b+1+num);
num=unique(b+1,b+1+num)-b-1;
for(int i=1;i<=n;i++) Change(i,1);
for(int i=1;i<=m;i++)
{
if(r[i])
{
printf("%d\n",b[Kth_pre(l[i],r[i],d[i])]);
}
else
{
Change(l[i],-1);
a[l[i]]=d[i];
Change(l[i],1);
}
}
}