1901: Zju2112 Dynamic Rankings|树状数组套主席树

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<vector>
#include<cstdio>
#include<queue>
#include<cmath>
#include<set>
#include<map>
#define ll long long
#define N 9000009
#define M 10010
using namespace std;
int sc()
{
int i=0,f=1; char c=getchar();
while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
return i*f;
}
int ch
[2],sum
;
int root[M],a[M];
int n,cnt,m,mx=1e9;
void insert(int pre,int &x,int l,int r,int v,int k)
{
if(!x)x=++cnt;
sum[x]=sum[pre]+k;
if(l==r)return ;
int mid=l+r>>1;
if(v<=mid)
ch[x][1]=ch[pre][1],
insert(ch[pre][0],ch[x][0],l,mid,v,k);
else
ch[x][0]=ch[pre][0],
insert(ch[pre][1],ch[x][1],mid+1,r,v,k);
}
void change(int x,int f,int k)
{
for(;x<=n;x+=x&-x)
insert(root[x],root[x],0,mx,f,k);
}
int query(int l,int r,int K)
{
int a[22],b[22],la=0,lb=0;
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
for(int i=l-1;i;i-=i&-i)a[++la]=root[i];
for(int i=r;i;i-=i&-i)b[++lb]=root[i];
int L=0,R=mx,n=max(la,lb);
while(L!=R)
{
int mid=L+R>>1,Sum=0;
for(int i=1;i<=n;i++)
Sum+=sum[ch[b[i]][0]]-sum[ch[a[i]][0]];
int e=(Sum<K);
if(e)K-=Sum,L=mid+1;else R=mid;
for(int i=1;i<=n;i++)
a[i]=ch[a[i]][e],b[i]=ch[b[i]][e];
}
return L;
}
int main()
{
n=sc(),m=sc();
for(int i=1;i<=n;i++)
a[i]=sc(),change(i,a[i],1);
while(m--)
{
char s[5]; scanf("%s",s);
if(s[0]=='Q')
{
int l=sc(),r=sc(),K=sc();
printf("%d\n",query(l,r,K));
}
else
{
int x=sc(),y=sc();
change(x,a[x],-1);
change(x,a[x]=y,1);
}
}
return 0;
}