[BZOJ1901]Dynamic Rankings（动态主席树）

题目描述

传送门

题解

动态主席树模板题，感谢hzwer的模板，自己改动了一点小地方。
这就是传说中的树套树？

代码

#include<algorithm>#include<iostream>#include<cstring>#include<cstdio>using namespace std;const int max_n=1e4+5;const int max_m=1e4+5;const int max_log=30;const int max_tree=3e6+5;char opt[10];int n,m,a,b,t,sz,N,cnt;int val[max_n],A[max_m],B[max_m],K[max_m],flag[max_m];int num[max_n*2],hash[max_n*2],root[max_n*2],L[max_log],R[max_log];int sum[max_tree],ls[max_tree],rs[max_tree];inline int find(int x){    int l=1,r=N,mid;    while (l<=r){        mid=(l+r)>>1;        if (hash[mid]<x) l=mid+1;        else r=mid-1;     }    return l;}inline void build(int last,int l,int r,int &now,int x,int v){    int mid=(l+r)>>1;    now=++sz;    sum[now]=sum[last]+v; ls[now]=ls[last]; rs[now]=rs[last];    if (l==r) return;    if (x<=mid) build(ls[last],l,mid,ls[now],x,v);    else build(rs[last],mid+1,r,rs[now],x,v);}inline int query(int l,int r,int k){    int mid=(l+r)>>1;    if (l==r) return l;    int suml=0,sumr=0;    for (int i=1;i<=a;++i) suml+=sum[ls[L[i]]];    for (int i=1;i<=b;++i) sumr+=sum[ls[R[i]]];    if (sumr-suml>=k){        for (int i=1;i<=a;++i) L[i]=ls[L[i]];        for (int i=1;i<=b;++i) R[i]=ls[R[i]];        return query(l,mid,k);    }    else{        for (int i=1;i<=a;++i) L[i]=rs[L[i]];        for (int i=1;i<=b;++i) R[i]=rs[R[i]];        return query(mid+1,r,k-(sumr-suml));    }}int main(){    scanf("%d%d",&n,&m);    for (int i=1;i<=n;++i) scanf("%d",&val[i]),num[++cnt]=val[i];    for (int i=1;i<=m;++i){        scanf("%s",opt);        if (opt[0]=='C'){            scanf("%d%d",&A[i],&B[i]);            num[++cnt]=B[i];        }        else{            flag[i]=1;            scanf("%d%d%d",&A[i],&B[i],&K[i]);        }    }    sort(num+1,num+cnt+1);    hash[++N]=num[1];    for (int i=2;i<=cnt;++i)      if (num[i]!=num[i-1]) hash[++N]=num[i];    for (int i=1;i<=n;++i){        t=find(val[i]);        for (int j=i;j<=n;j+=j&(-j))          build(root[j],1,N,root[j],t,1);    }    for (int i=1;i<=m;++i)      if (flag[i]){        a=0,b=0,A[i]--;        for (int j=A[i];j>=1;j-=j&(-j)) L[++a]=root[j];        for (int j=B[i];j>=1;j-=j&(-j)) R[++b]=root[j];        printf("%d\n",hash[query(1,N,K[i])]);      }      else{        t=find(val[A[i]]);        for (int j=A[i];j<=n;j+=j&(-j)) build(root[j],1,N,root[j],t,-1);        val[A[i]]=B[i];        t=find(val[A[i]]);        for (int j=A[i];j<=n;j+=j&(-j)) build(root[j],1,N,root[j],t,1);      }}