【bzoj1901】dynamic ranking（带修改主席树）

传送门（权限）

传送门（非权限）

花了一晚上总算把代码调好了……才知道待修改主席树怎么操作……

然而还是一知半解orz……

先说说我的理解吧

我们一般建主席树的时候都是直接在序列上建的

但是如果有修改操作怎么办？

因为主席树维护的是前缀和

而树状数组刚好支持待修改前缀和

所以我们可以将主席树和树状数组一起使用

树状数组实际指向主席树上的节点

修改和查询操作用树状数组遍历，实则修改或查询主席树上的节点

 //minamoto
 #include<bits/stdc++.h>
 #define N 10005
 using namespace std;
 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
 char buf[<<],*p1=buf,*p2=buf;
 inline int read(){
     #define num ch-'0'
     char ch;bool flag=;int res;
     while(!isdigit(ch=getc()))
     (ch=='-')&&(flag=true);
     for(res=num;isdigit(ch=getc());res=res*+num);
     (flag)&&(res=-res);
     #undef num
     return res;
 }
 char obuf[<<],*o=obuf;
 void print(int x){
     if(x>) print(x/);
     *o++=x%+;
 }
 inline int lowbit(int x){return x&(-x);}
 int sum[N*],L[N*],R[N*];
 int xx[N],yy[N],rt[N],a[N],b[N<<],ca[N],cb[N],cc[N];
 int n,q,m,cnt=,totx,toty;
 void update(int last,int &now,int l,int r,int x,int v){
     sum[now=++cnt]=sum[last]+v;
     L[now]=L[last],R[now]=R[last];
     if(l==r) return;
     int mid=(l+r)>>;
     if(x<=mid) update(L[last],L[now],l,mid,x,v);
     else update(R[last],R[now],mid+,r,x,v);
 }
 int query(int l,int r,int q){
     if(l==r) return l;
     int x=,mid=(l+r)>>;
     for(int i=;i<=totx;++i) x-=sum[L[xx[i]]];
     for(int i=;i<=toty;++i) x+=sum[L[yy[i]]];
     if(q<=x){
         for(int i=;i<=totx;++i) xx[i]=L[xx[i]];
         for(int i=;i<=toty;++i) yy[i]=L[yy[i]];
         return query(l,mid,q);
     }
     else{
         for(int i=;i<=totx;++i) xx[i]=R[xx[i]];
         for(int i=;i<=toty;++i) yy[i]=R[yy[i]];
         return query(mid+,r,q-x);
     }
 }
 void add(int x,int y){
     int k=lower_bound(b+,b++m,a[x])-b;
     for(int i=x;i<=n;i+=lowbit(i)) update(rt[i],rt[i],,m,k,y);
 }
 int main(){
     //freopen("testdata.in","r",stdin);
     n=read(),q=read();
     for(int i=;i<=n;++i)
     b[++m]=a[i]=read();
     for(int i=;i<=q;++i){
         char ch;
         while(!isupper(ch=getc()));
         ca[i]=read(),cb[i]=read();
         if(ch=='Q') cc[i]=read();else b[++m]=cb[i];
     }
     sort(b+,b++m);
     m=unique(b+,b++m)-b-;
     for(int i=;i<=n;++i) add(i,);
     for(int i=;i<=q;++i){
         if(cc[i]){
             totx=toty=;
             for(int j=ca[i]-;j;j-=lowbit(j)) xx[++totx]=rt[j];
             for(int j=cb[i];j;j-=lowbit(j)) yy[++toty]=rt[j];
             print(b[query(,m,cc[i])]),*o++='\n';
         }
         else{add(ca[i],-),a[ca[i]]=cb[i],add(ca[i],);}
     }
     fwrite(obuf,o-obuf,,stdout);
     return ;
 }

然而我太菜了不会树套树和整体二分……

这里是zcysky大佬的树套树

这里是will大爷的整体二分