ZOJ---2112：Dynamic Rankings【动态主席树】

题目：

ZOJ---2112：Dynamic Rankings

题意：

求区间第 K 大，但有单点修改操作

笔记:

主席树的本质就是权值线段树+前缀和，静态主席树也是按照前缀和的思想建的权值线段树；如果现在有修改操作，那么就得修改这个点后面所有的权值线段树；考虑到查询的也是前缀和，于是就引入树状数组来加速这个过程，实际上就是牺牲查询加速修改

 核心：

现在树状数组每个节点都是一颗权值线段树，它维护了区间【x-lowbit[x]+1，x】的信息，并且树状数组只处理修改的信息，只需要在权值线段树对应的位置加减 1 即可

查询的时候和静态查询一样的方式，只不过现在要加入修改的信息，也就是要查询树状数组上这一层（静态主席树同一层）的前缀和，就可以得到区间所有数的个数

学习自：动态主席树 

代码：

#include <bits/stdc++.h>

#define lowbit(x) x&(-x)
using namespace std;
const int maxn = 1e5+54;
int c[maxn],a[maxn],n,m,Case,cnt,tot;
int root[maxn],bit[maxn],nxt[2][maxn];
/*
    root[] : 静态主席树的根
    bit[]  : 树状数组的根
    nxt[][]: 记录树状数组到达同一层的标号，要同时记录qL-1，qR
*/
struct node{
    int f,l,r,k;
}q[maxn];
struct tree{
    int l,r,sum;
}T[maxn*80];
inline int getid(int x){
    return lower_bound(c+1,c+cnt,x)-c;
}
void updata(int l,int r,int &x,int y,int val,int w){
    T[++tot] = T[y]; T[tot].sum += w; x = tot;
    if(l == r) return ;
    int mid = (l+r) >> 1;
    if(val > mid) updata(mid+1,r,T[x].r,T[y].r,val,w);
    else updata(l,mid,T[x].l,T[y].l,val,w);
}
void change(int x,int w){
    int val = getid(a[x]);
    while(x < cnt){
        updata(1,cnt,bit[x],bit[x],val,w);
        x += lowbit(x);
    }
}
int persum(int x,int t){
    int res = 0;
    while(x){
        res += T[T[nxt[t][x]].l].sum;
        x -= lowbit(x);
    }
    return res;
}
void query(int l,int r,int L,int R,int x,int y,int k,int &ans){
    if(l == r) {ans = l; return ;}
    int num = persum(R,0)-persum(L,1)+T[T[x].l].sum-T[T[y].l].sum;    //求当前区间左区间的和
    if(num >= k){
        for(int i = R;i > 0;i-=lowbit(i)) nxt[0][i] = T[nxt[0][i]].l;
        for(int i = L;i > 0;i-=lowbit(i)) nxt[1][i] = T[nxt[1][i]].l;
        query(l,(l+r)>>1,L,R,T[x].l,T[y].l,k,ans);
    }else{
        for(int i = R;i > 0;i-=lowbit(i)) nxt[0][i] = T[nxt[0][i]].r;
        for(int i = L;i > 0;i-=lowbit(i)) nxt[1][i] = T[nxt[1][i]].r;
        query(((l+r)>>1)+1,r,L,R,T[x].r,T[y].r,k-num,ans);
    }
}
int main(){
    scanf("%d",&Case);
    while(Case--){
        scanf("%d %d",&n,&m); tot = 0; cnt = 1;
        for(int i = 1;i <= n; ++i) scanf("%d",a+i),c[cnt++]=a[i];
        for(int i = 0;i < m; ++i){
            char op; cin >> op;
            if(op == 'Q'){
                q[i].f = 1;
                scanf("%d %d %d",&q[i].l,&q[i].r,&q[i].k);
            }else{
                q[i].f = 0;
                scanf("%d %d",&q[i].l,&q[i].r);
                c[cnt++] = q[i].r;
            }
        }
        sort(c+1,c+cnt); cnt = unique(c+1,c+cnt)-c;
        for(int i = 1;i <= n; ++i) updata(1,cnt,root[i],root[i-1],getid(a[i]),1),bit[i]=0;
        for(int i = 0;i < m; ++i){
            if(q[i].f == 0){
                change(q[i].l,-1); 
                a[q[i].l] = q[i].r;
                change(q[i].l,1);
            }else{
                int ans;
                for(int j = q[i].r;j > 0;j -= lowbit(j)) nxt[0][j] = bit[j];
                for(int j = q[i].l-1;j > 0;j-=lowbit(j)) nxt[1][j] = bit[j];
                query(1,cnt,q[i].l-1,q[i].r,root[q[i].r],root[q[i].l-1],q[i].k,ans);
                printf("%d\n",c[ans]);
            }
        }
    }
    return 0;
}