【BZOJ】1901: Zju2112 Dynamic Rankings

【题意】带修改的查询区间第k小

【算法】树状数组套可持久化线段树

【题解】对于树状数组上的每个节点，维护可持久化权值线段树（节点为权值），从而达到查询前缀和的目的。

对于每次修改，在待修改线段树基础上运用可持久化性质来修改，先删除原数字，再加入新数字。

注意记录修改后的原数字，方便后来删除。

★注意区分权值范围和数组范围的区别！！！

复杂度O(n log²n)。

注意点全部标注在代码中，细节巨多。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cctype>
using namespace std;
int read()
{
    char c;int s=,t=;
    while(!isdigit(c=getchar()))if(c=='-')t=-;
    do{s=s*+c-'';}while(isdigit(c=getchar()));
    return s*t;
}
const int maxn=;
struct tree{int l,r,sum;}t[];//空间开大！！！100w都过不去啊。。。
char s[];
int cnt1,cnt2,a1[maxn],a2[maxn],a[maxn],b1[maxn],b2[maxn],b3[maxn],n,m,root[maxn],tot,c[maxn*],sz=;

void insert(int l,int r,int x,int &y,int v,int c){
    y=++sz;//设置新节点！！！
    t[y]=t[x];t[y].sum+=c;
    if(l==r)return;
    int mid=(l+r)>>;
    if(v<=mid)insert(l,mid,t[x].l,t[y].l,v,c);
    else insert(mid+,r,t[x].r,t[y].r,v,c);
}
int ask(int l,int r,int v){
    if(l==r)return l;
    int sum=;
    for(int i=;i<=cnt1;i++)sum-=t[t[a1[i]].l].sum;
    for(int i=;i<=cnt2;i++)sum+=t[t[a2[i]].l].sum;
    int mid=(l+r)>>;
    if(sum>=v){
        for(int i=;i<=cnt1;i++)a1[i]=t[a1[i]].l;
        for(int i=;i<=cnt2;i++)a2[i]=t[a2[i]].l;
        return ask(l,mid,v);
    }
    else{
        for(int i=;i<=cnt1;i++)a1[i]=t[a1[i]].r;
        for(int i=;i<=cnt2;i++)a2[i]=t[a2[i]].r;
        return ask(mid+,r,v-sum);//转入右边要减去左边的
    }
}

int lowbit(int x){return x&(-x);}
void modify(int x,int k,int p){for(int i=x;i<=n;i+=lowbit(i))insert(,tot,root[i],root[i],k,p);}
//分清，tot是权值线段树范围，n是数组范围
int main(){
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++){
        a[i]=read();
        c[++tot]=a[i];
    }
    for(int i=;i<=m;i++){
        scanf("%s%d%d",s,&b1[i],&b2[i]);
        if(s[]=='Q')b3[i]=read();
        else c[++tot]=b2[i];
    }
    sort(c+,c+tot+);
    tot=unique(c+,c+tot+)-c-;
    for(int i=;i<=n;i++)a[i]=lower_bound(c+,c+tot+,a[i])-c;
    for(int i=;i<=m;i++)if(!b3[i])b2[i]=lower_bound(c+,c+tot+,b2[i])-c;
    for(int i=;i<=n;i++)modify(i,a[i],);
    for(int i=;i<=m;i++){
        if(b3[i]){
            cnt1=;cnt2=;
            for(int j=b1[i]-;j>=;j-=lowbit(j))a1[++cnt1]=root[j];
            for(int j=b2[i];j>=;j-=lowbit(j))a2[++cnt2]=root[j];//用根啊老铁
            printf("%d\n",c[ask(,tot,b3[i])]);//询问给的是离散化后的值，要还原
        }
        else{
            modify(b1[i],a[b1[i]],-);
            a[b1[i]]=b2[i];
            modify(b1[i],b2[i],);
        }
    }
    return ;
}