P2617 Dynamic Rankings (动态区间第K大，树状数组套动态开点线段树)

题目描述

给定一个含有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≤100000)，m(1≤m≤100000)。分别表示序列的长度和指令的个数。

第二行有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。

输出格式

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

输入输出样例

输入 #1

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

输出 #1

3
6

说明/提示

对于所有数据，m,n≤100000

思路：

主席树如果修改x位置上的数，则x之后的全部都要修改，直接爆了
可以用树状数组维护前缀和，这样只需要修改logn个位置

查询时，仍然是R位置减去L-1位置，但是从两棵线段树作差变为log棵线段树与log棵线段树作差，查询左子树(右子树)的时候，log个节点一起跳到左子树(或者右子树)

code：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
using namespace std;
const int maxm=1e5+5;
struct Query{//query
    int l,r,k;
}q[maxm];
struct Tree{//tree
    int lc,rc,cnt;
}a[maxm<<8];
int rt[maxm],cnt;//存根
int p[maxm];//原数组
int xx[maxm<<1],num;//离散化
int L[maxm],c1;//树状数组的1-(l-1)的rt,因为lowbit所以这个数组也可以不用开很大
int R[maxm],c2;//树状数组的1-r的rt
int n,m;
int lowbit(int i){
    return i&-i;
}
void update(int &k,int l,int r,int x,int val){
    if(!k)k=++cnt;
    a[k].cnt+=val;
    if(l==r)return ;
    int mid=(l+r)/2;
    if(x<=mid)update(a[k].lc,l,mid,x,val);
    else update(a[k].rc,mid+1,r,x,val);
}
void add(int x,int val){
    int k=lower_bound(xx+1,xx+1+num,p[x])-xx;
    for(int i=x;i<=n;i+=lowbit(i)){
        update(rt[i],1,num,k,val);
    }
}
int ask(int l,int r,int k){
    if(l==r)return xx[l];
    int mid=(l+r)/2;
    int res=0;
    for(int i=1;i<=c2;i++){//加上1-r
        res+=a[a[R[i]].lc].cnt;
    }
    for(int i=1;i<=c1;i++){//减去1-(l-1)
        res-=a[a[L[i]].lc].cnt;
    }
    if(res>=k){//左子树数量大于k，说明第k大在左子树
        for(int i=1;i<=c1;i++){//更新1-(l-1)
            L[i]=a[L[i]].lc;
        }
        for(int i=1;i<=c2;i++){//更新1-r
            R[i]=a[R[i]].lc;
        }
        return ask(l,mid,k);
    }else{//否则在右子树
        for(int i=1;i<=c1;i++){//更新1-(l-1)
            L[i]=a[L[i]].rc;
        }
        for(int i=1;i<=c2;i++){//更新1-r
            R[i]=a[R[i]].rc;
        }
        return ask(mid+1,r,k-res);
    }
}
int kth(int l,int r,int k){
    l--;
    c1=c2=0;
    for(int i=l;i;i-=lowbit(i)){//树状数组1-(l-1)的rt
        L[++c1]=rt[i];
    }
    for(int i=r;i;i-=lowbit(i)){//树状数组1-r的rt
        R[++c2]=rt[i];
    }
    return ask(1,num,k);
}
signed main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&p[i]);
        xx[++num]=p[i];
    }
    for(int i=1;i<=m;i++){
        char s[5];
        scanf("%s",s);
        scanf("%d%d",&q[i].l,&q[i].r);
        if(s[0]=='Q'){//ask [l,r] kth
            scanf("%d",&q[i].k);
        }else{//change p[l] to r
            q[i].k=0;//用k等于0表示change
            xx[++num]=q[i].r;//所有数据加入离散化
        }
    }
    sort(xx+1,xx+1+num);
    num=unique(xx+1,xx+1+num)-xx-1;
    for(int i=1;i<=n;i++){
        add(i,1);
    }
    for(int i=1;i<=m;i++){
        if(q[i].k){//ask
            printf("%d\n",kth(q[i].l,q[i].r,q[i].k));
        }else{//change
            add(q[i].l,-1);
            p[q[i].l]=q[i].r;
            add(q[i].l,1);
        }
    }
    return 0;
}