[BZOJ1901]Dynamic Rankings

Description

给定一个含有n个数的序列a[1],a[2],a[3]……a[n]，程序必须回答这样的询问：对于给定的i,j,k，在a[i],a[i+1

],a[i+2]……a[j]中第k小的数是多少(1≤k≤j-i+1)，并且，你可以改变一些a[i]的值，改变后，程序还能针对改

变后的a继续回答上面的问题。

Input

第一行有两个正整数n(1≤n≤10000)，m(1≤m≤10000)。

分别表示序列的长度和指令的个数。

第二行有n个数，表示a[1],a[2]……a[n]，这些数都小于10^9。

接下来的m行描述每条指令

每行的格式是下面两种格式中的一种。 

Q i j k 或者 C i t 

Q i j k （i,j,k是数字，1≤i≤j≤n, 1≤k≤j-i+1）

表示询问指令，询问a[i]，a[i+1]……a[j]中第k小的数。

C i t (1≤i≤n，0≤t≤10^9)表示把a[i]改变成为t

m,n≤10000

Output

对于每一次询问，你都需要输出他的答案，每一个输出占单独的一行。

Sample Input

5 3
3 2 1 4 7
Q 1 4 3
C 2 6
Q 2 5 3

Sample Output

3
6

 

第一次写树状数组套主席树，还是很好写的，思路也非常好理解

代码：

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define M 20010
 #define ls L[node]
 #define rs R[node]
 using namespace std;
 int n,m,sz,cnt,tx,ty;
 int a[M<<],b[M<<],px[M<<],py[M<<],ql[M<<],qr[M<<],qk[M<<];
 int rt[M<<],val[M<<],L[M<<],R[M<<];
 void insert(int &node,int l,int r,int x,int v) {
     if(!node) node=++sz;val[node]+=v;
     if(l==r) return;
     int mid=(l+r)/;
     if(x<=mid) insert(ls,l,mid,x,v);
     else insert(rs,mid+,r,x,v);
 }
 void add(int loc,int v) {
     int p=lower_bound(b+,b++cnt,a[loc])-b;
     for(int i=loc;i<=n;i+=(i&-i))
         insert(rt[i],,cnt,p,v);
 }
 int query(int l,int r,int k) {
     if(l==r) return l;
     int sum=,mid=(l+r)/;
     for(int i=;i<=tx;i++) sum-=val[L[px[i]]];
     for(int i=;i<=ty;i++) sum+=val[L[py[i]]];
     if(k<=sum) {
         for(int i=;i<=tx;i++) px[i]=L[px[i]];
         for(int i=;i<=ty;i++) py[i]=L[py[i]];
         return query(l,mid,k);
     }
     else {
         k-=sum;
         for(int i=;i<=tx;i++) px[i]=R[px[i]];
         for(int i=;i<=ty;i++) py[i]=R[py[i]];
         return query(mid+,r,k);
     }
 }
 int main() {
     scanf("%d%d",&n,&m);cnt=n;
     for(int i=;i<=n;i++) scanf("%d",&a[i]),b[i]=a[i];
     for(int i=;i<=m;i++) {
         char s[];scanf("%s",s);
         scanf("%d%d",&ql[i],&qr[i]);
         if(s[]=='Q') scanf("%d",&qk[i]);
         else b[++cnt]=qr[i];
     }
     sort(b+,b++cnt);
     cnt=unique(b+,b++cnt)-b-;
     for(int i=;i<=n;i++) add(i,);
     for(int i=;i<=m;i++) {
         if(qk[i]) {
             tx=ty=;
             for(int j=ql[i]-;j;j-=(j&-j)) px[++tx]=rt[j];
             for(int j=qr[i];j;j-=(j&-j)) py[++ty]=rt[j];
             printf("%d\n",b[query(,cnt,qk[i])]);
         }
         else {
             add(ql[i],-);
             a[ql[i]]=qr[i];
             add(ql[i],);
         }
     }
     return ;
 }