题目大意:区间k大,支持单点修改
题解:带修主席树模板
我的收获:233
#include <bits/stdc++.h>
using namespace std;
#define N 10005
int n,m,top,cnt,tot;
int v[N],num[N<<1];
int A[N],B[N],K[N],flag[N];
int root[N],L[55],R[55];
inline int pos(int x){return lower_bound(num+1,num+1+cnt,x)-num;}
struct ChairmanTree{
int sum[N*300],tl[N*300],tr[N*300];
void upd(int k,int l,int r,int v,int x,int &y)
{
y=++tot,sum[y]=sum[x]+v,tl[y]=tl[x],tr[y]=tr[x];
if(l==r) return ;
int mid=l+r>>1;
if(k<=mid) upd(k,l,mid,v,tl[x],tl[y]);
if(k>mid) upd(k,mid+1,r,v,tr[x],tr[y]);
}
int query(int l,int r,int k)
{
if(l==r) return l;
int mid=l+r>>1,suml(0),sumr(0);
for(int i=1;i<=L[0];i++) suml+=sum[tl[L[i]]];
for(int i=1;i<=R[0];i++) sumr+=sum[tl[R[i]]];
if(sumr-suml>=k){
for(int i=1;i<=L[0];i++) L[i]=tl[L[i]];
for(int i=1;i<=R[0];i++) R[i]=tl[R[i]];
return query(l,mid,k);
}
else{
for(int i=1;i<=L[0];i++) L[i]=tr[L[i]];
for(int i=1;i<=R[0];i++) R[i]=tr[R[i]];
return query(mid+1,r,k-(sumr-suml));
}
}
}T;
struct Binary_indexed_tree{
void add(int x,int wr,int val){
for(;x<=cnt;x+=x&-x) T.upd(wr,1,cnt,val,root[x],root[x]);
}
int ask(int x,int y,int z){
L[0]=R[0]=0;
for(;x;x-=x&-x) L[++L[0]]=root[x];
for(;y;y-=y&-y) R[++R[0]]=root[y];
return T.query(1,cnt,z);
}
}C;
void work()
{
for(int i=1;i<=n;i++) C.add(i,pos(v[i]),1);
for(int i=1;i<=m;i++){
if(!flag[i]) printf("%d\n",num[C.ask(A[i]-1,B[i],K[i])]);
else{
C.add(A[i],pos(v[A[i]]),-1);
v[A[i]]=B[i];
C.add(A[i],pos(v[A[i]]),1);
}
}
}
void init()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&v[i]),num[++top]=v[i];
for(int i=1;i<=m;i++)
{
char opt[3];scanf("%s%d%d",opt,&A[i],&B[i]);
if(opt[0]=='Q') scanf("%d",&K[i]),flag[i]=0;
else num[++top]=B[i],flag[i]=1;
}
sort(num+1,num+top+1);cnt=unique(num+1,num+top+1)-num-1;
}
int main()
{
init();
work();
return 0;
}