1901: Zju2112 Dynamic Rankings

题目链接

题目大意：区间k大，支持单点修改

题解：带修主席树模板

我的收获：233

#include <bits/stdc++.h>
using namespace std;

#define N 10005

int n,m,top,cnt,tot;
int v[N],num[N<<1];
int A[N],B[N],K[N],flag[N];
int root[N],L[55],R[55];

inline int pos(int x){return lower_bound(num+1,num+1+cnt,x)-num;}

struct ChairmanTree{

    int sum[N*300],tl[N*300],tr[N*300];

    void upd(int k,int l,int r,int v,int x,int &y)
    {
        y=++tot,sum[y]=sum[x]+v,tl[y]=tl[x],tr[y]=tr[x];
        if(l==r) return ;
        int mid=l+r>>1;
        if(k<=mid) upd(k,l,mid,v,tl[x],tl[y]);
        if(k>mid) upd(k,mid+1,r,v,tr[x],tr[y]);
    }

    int query(int l,int r,int k)
    {
        if(l==r) return l;
        int mid=l+r>>1,suml(0),sumr(0);
        for(int i=1;i<=L[0];i++) suml+=sum[tl[L[i]]];
        for(int i=1;i<=R[0];i++) sumr+=sum[tl[R[i]]];        
        if(sumr-suml>=k){
            for(int i=1;i<=L[0];i++) L[i]=tl[L[i]];
            for(int i=1;i<=R[0];i++) R[i]=tl[R[i]];
            return query(l,mid,k);
        }
        else{
            for(int i=1;i<=L[0];i++) L[i]=tr[L[i]];
            for(int i=1;i<=R[0];i++) R[i]=tr[R[i]];
            return query(mid+1,r,k-(sumr-suml));
        }
    }

}T;

struct Binary_indexed_tree{

    void add(int x,int wr,int val){
        for(;x<=cnt;x+=x&-x) T.upd(wr,1,cnt,val,root[x],root[x]);
    }

    int ask(int x,int y,int z){
        L[0]=R[0]=0;
        for(;x;x-=x&-x) L[++L[0]]=root[x];
        for(;y;y-=y&-y) R[++R[0]]=root[y];
        return T.query(1,cnt,z);
    }

}C;

void work()
{
    for(int i=1;i<=n;i++) C.add(i,pos(v[i]),1);
    for(int i=1;i<=m;i++){
        if(!flag[i]) printf("%d\n",num[C.ask(A[i]-1,B[i],K[i])]);
        else{
            C.add(A[i],pos(v[A[i]]),-1);
            v[A[i]]=B[i];
            C.add(A[i],pos(v[A[i]]),1);
        }
    }
}

void init()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&v[i]),num[++top]=v[i];
    for(int i=1;i<=m;i++)
    {
        char opt[3];scanf("%s%d%d",opt,&A[i],&B[i]);
        if(opt[0]=='Q') scanf("%d",&K[i]),flag[i]=0;
        else num[++top]=B[i],flag[i]=1;
    }
    sort(num+1,num+top+1);cnt=unique(num+1,num+top+1)-num-1;
}

int main()
{
    init();
    work();
    return 0;
}