BZOJ.1901.Dynamic Rankings(树状数组套主席树(动态主席树))

题目链接 BZOJ
洛谷

区间第k小，我们可以想到主席树。然而这是静态的，怎么支持修改？
静态的主席树是利用前缀和+差分来求解的，那么对于每个位置上的每棵树看做一个点，拿树状数组更新。
还是树状数组的过程，区间加时，每到一个位置在这棵主席树中插入这个数。
查询时，将所有询问要访问到的主席树存下来，delta为所有存下的树的和的差值；改变节点时所有的主席树访问节点都变。
每次树状数组会访问logn棵树，每棵树改变logn个点。时间空间复杂度都为 \(O(nlog^2n)\).

线段树套平衡树见这.
整体二分见这.

//27068kb   364ms
#include <cstdio>
#include <cctype>
#include <algorithm>
//#define gc() getchar()
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
#define lb(x) (x)&-(x)
const int N=1e4+5,S=N*220,MAXIN=1e5;

int n,Q,A[N],cnt,ref[N<<1];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Operation{ //离线离散化 
    int l,r,K;//l=0:Modify r:pos K:val
    Operation() {}
    Operation(int l,int r,int k):l(l),r(r),K(k) {}
}q[N];
inline int read()
{
    int now=0,f=1;register char c=gc();
    for(;!isdigit(c);c=gc()) if(c=='-') f=-1;
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now*f;
}
namespace T
{
    #define lson son[rt][0]
    #define rson son[rt][1]

    int tot,son[S][2],sz[S],root[N],totl,totr,ql[N],qr[N];
    void Insert(int &rt,int l,int r,int p,int v)
    {
        if(!rt) rt=++tot;
        sz[rt]+=v;//既然直接在自己这棵树上改，不需要再复制很多重复节点。
        if(l<r){
            int m=l+r>>1;
            if(p<=m) Insert(lson,l,m,p,v);
            else Insert(rson,m+1,r,p,v);
        }
    }
    void Modify(int p,int v,int delta){
        while(p<=n)
            Insert(root[p],1,cnt,v,delta),p+=lb(p);
    }
    int Query(int l,int r,int k)
    {
        if(l==r) return ref[l];
        int delta=0;
        for(int i=1; i<=totl; ++i) delta-=sz[son[ql[i]][0]];
        for(int i=1; i<=totr; ++i) delta+=sz[son[qr[i]][0]];
        if(delta>=k){
            for(int i=1; i<=totl; ++i) ql[i]=son[ql[i]][0];
            for(int i=1; i<=totr; ++i) qr[i]=son[qr[i]][0];
            return Query(l,l+r>>1,k);
        }
        else{
            for(int i=1; i<=totl; ++i) ql[i]=son[ql[i]][1];
            for(int i=1; i<=totr; ++i) qr[i]=son[qr[i]][1];
            return Query((l+r>>1)+1,r,k-delta);
        }
    }
    int Kth(int l,int r,int k)
    {//别忘l-1.
        totl=totr=0;
        for(--l; l; l^=lb(l)) ql[++totl]=root[l];//存根。
        for(; r; r^=lb(r)) qr[++totr]=root[r];
        return Query(1,cnt,k);
    }
}
inline char Get(){
    char c=gc();while(c!='Q'&&c!='C') c=gc();
    return c;
}
int Find(int x)
{
    int l=1,r=cnt,mid;
    while(l<r)
        if(ref[mid=l+r>>1]<x) l=mid+1;
        else r=mid;
    return l;
}

int main()
{
    n=read(),Q=read();
    for(int i=1; i<=n; ++i) ref[i]=A[i]=read();
    int t=n;
    for(int l,r,k,i=1; i<=Q; ++i)
        if(Get()=='Q') l=read(),r=read(),k=read(), q[i]=Operation(l,r,k);
        else r=read(),k=read(), q[i]=Operation(0,r,k),ref[++t]=k;

    std::sort(ref+1,ref+1+t), cnt=1;
    for(int i=2; i<=t; ++i)
        if(ref[i]!=ref[i-1]) ref[++cnt]=ref[i];

    for(int i=1; i<=n; ++i) T::Modify(i,A[i]=Find(A[i]),1);
    for(int i=1; i<=Q; ++i)
        if(!q[i].l) T::Modify(q[i].r,A[q[i].r],-1), T::Modify(q[i].r,A[q[i].r]=Find(q[i].K),1);
        else printf("%d\n",T::Kth(q[i].l,q[i].r,q[i].K));
    return 0;
}

转载于:https://www.cnblogs.com/SovietPower/p/8653956.html