【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继续回答上面的问题。你需要编一个这样的程序，从输入文件中读入序列a，然后读入一系列的指令，包括询问指令和修改指令。对于每一个询问指令，你必须输出正确的回答。 第一行有两个正整数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。

Input

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

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

HINT

20%的数据中，m,n≤100; 40%的数据中，m,n≤1000; 100%的数据中，m,n≤10000。

题解：发现不会动态主席，就学习了一下。

           大致就是树状数组套可持久化线段树。

           静态主席树维护的是前缀和，这个只需要维护节点信息，前缀和用树状数组维护即可。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 3000000
#define M 20010
int sz,ls[N],rs[N],hash[M],sum[N],root[M],n,m,v[M],A[M],B[M],K[M],tot(1),f[M],L[M],R[M],num;
char s[10];
using namespace std;
int find(int x){
  int l=1,r=tot;
  while (l<=r){
    int mid=(l+r)>>1;
    if (x<=hash[mid]) r=mid-1;
    else l=mid+1;
  }
 return l;
}
int lowbit(int x){return x&(-x);}
void updata(int last,int l,int r,int &rt,int x,int v){
  rt=++sz;sum[rt]=sum[last]+v;ls[rt]=ls[last];rs[rt]=rs[last];
  if (l==r) return;int mid=l+r>>1;
  if (x<=mid) updata(ls[last],l,mid,ls[rt],x,v);
  else updata(rs[last],mid+1,r,rs[rt],x,v);	
}
int query(int l,int r,int k){
  int sum1(0),sum2(0),mid=l+r>>1;
  if (l==r) return l;
  for (int i=1;i<=L[0];i++) sum1+=sum[ls[L[i]]];
  for (int i=1;i<=R[0];i++) sum2+=sum[ls[R[i]]];
  if (sum2-sum1>=k){
   for (int i=1;i<=L[0];i++) L[i]=ls[L[i]];
   for (int i=1;i<=R[0];i++) R[i]=ls[R[i]];
   return query(l,mid,k);
  }
  else{
   for (int i=1;i<=L[0];i++) L[i]=rs[L[i]];
   for (int i=1;i<=R[0];i++) R[i]=rs[R[i]];
   return query(mid+1,r,k-(sum2-sum1));
  }
}
int main(){
 scanf("%d%d",&n,&m);
 for (int i=1;i<=n;i++)	scanf("%d",&v[i]),hash[++num]=v[i];
 for (int i=1;i<=m;i++){
   scanf("%s%d%d",&s,&A[i],&B[i]);
   if (s[0]=='Q') scanf("%d",&K[i]),f[i]=0;
   else hash[++num]=B[i],f[i]=1;
 }
 sort(hash+1,hash+num+1);
 for (int i=2;i<=num;i++)
  if (hash[i]!=hash[i-1]) hash[++tot]=hash[i];
// for (int i=1;i<=tot;i++) cout<<hash[i]<<endl;
 for (int i=1;i<=n;i++){
   int t=find(v[i]);
   for (int j=i;j<=n;j+=lowbit(j))
    updata(root[j],1,tot,root[j],t,1);
 }
 for (int i=1;i<=m;i++){
  if (f[i]){
  	int t=find(v[A[i]]);
    for (int j=A[i];j<=n;j+=lowbit(j))
      updata(root[j],1,tot,root[j],t,-1);
    v[A[i]]=B[i];t=find(B[i]);
    for (int j=A[i];j<=n;j+=lowbit(j))
     updata(root[j],1,tot,root[j],t,1);
  }
  else{
  	A[i]--;R[0]=L[0]=0;
  	for (int j=A[i];j;j-=lowbit(j)) L[++L[0]]=root[j];
  	for (int j=B[i];j;j-=lowbit(j)) R[++R[0]]=root[j];
    printf("%d\n",hash[query(1,tot,K[i])]);
  }
 }
}