洛谷P2617 Dynamic Ranking 树状数组套主席树

#include<stdio.h>#include<ctype.h>#include<algorithm>#define lowbit(x) (x&-x)#define N 10005using namespace std;int n,m,a[N],b[N*2],s[N][3],len;char c;int rt[N],val[N*400],lch[N*400],rch[N*400],cnt;int L[20],R[20],llen,rlen;void update(int &u,int l,int r,int k,int v) {if(!u) u=++cnt;val[u]+=v;if(l<r) {int mid=l+r>>1;if(k<=mid) update(lch[u],l,mid,k,v);else update(rch[u],mid+1,r,k,v);}}void Update(int x,int v) {int k=lower_bound(b+1,b+1+len,a[x])-b;for(; x<=n; x+=lowbit(x)) update(rt[x],1,len,k,v);}int query(int l,int r,int k) {if(l==r) return l;int mid=l+r>>1, tmp=0;for(int i=0; i<rlen; i++) tmp+=val[lch[R[i]]];for(int i=0; i<llen; i++) tmp-=val[lch[L[i]]];if(k<=tmp) {for(int i=0; i<rlen; i++) R[i]=lch[R[i]];for(int i=0; i<llen; i++) L[i]=lch[L[i]];return query(l,mid,k);} else {for(int i=0; i<rlen; i++) R[i]=rch[R[i]];for(int i=0; i<llen; i++) L[i]=rch[L[i]];return query(mid+1,r,k-tmp);}}int Query(int l,int r,int k) {llen=0, rlen=0;for(int i=r; i; i-=lowbit(i)) R[rlen++]=rt[i];for(int i=l-1; i; i-=lowbit(i)) L[llen++]=rt[i];return query(1,len,k);}int main() {scanf("%d%d",&n,&m);len=n;for(int i=1; i<=n; i++) {scanf("%d",&a[i]);b[i]=a[i];}for(int i=1; i<=m; i++) {while(isspace(c=getchar()));scanf("%d%d",&s[i][0],&s[i][1]);if(c=='Q') scanf("%d",&s[i][2]);else b[++len]=s[i][1], s[i][2]=-1;}sort(b+1,b+1+len);len=unique(b+1,b+1+len)-b-1;for(int i=1; i<=n; i++) Update(i,1);for(int i=1; i<=m; i++) {if(~s[i][2]) {printf("%d\n",b[Query(s[i][0],s[i][1],s[i][2])]);} else {Update(s[i][0],-1);a[s[i][0]]=s[i][1];Update(s[i][0],1);}}}