写挂了好多次,也不太理解
带修改的情况如果还用(权值线段树)主席树,每个点(root[i])维护前缀的某值出现次数,
每次修改需要对此点后面所有主席树修改,共$nlogn$个
这里用树状数组维护前缀和,平时的树状数组实际上每个点维护的是原序列$[i,i-lowbit(i)+1]$这些值,这里也让每个树状数组维护这样一个区间的信息,而不是前缀的,这样修改一个点的值只会影响$logn$棵主席树,查询也是$logn$棵主席树的信息加起来就是整个区间的信息,每次修改存下所有需要修改/查询的主席树,同步进行加减/向下跳,这样每次修改的复杂度就是$log^2 n$的
主席树真的好慢...跑了12s,动态区间第k大还有整体二分等做法
(我竟然连树状数组都不理解...还看了好久树状数组)
#include<bits/stdc++.h> #define mid (l+r>>1) #define lbt(x) (x&-x) using namespace std; const int maxn=100009; inline int read(){ int ret=0,fix=1;char ch; while(!isdigit(ch=getchar()))fix=ch=='-'?-1:fix; do ret=(ret<<1)+(ret<<3)+ch-'0'; while(isdigit(ch=getchar())); return fix*ret; } struct qqq{ int tp,l,r,x; }qu[maxn]; int n,m,tot,cnt; int hsh[maxn<<1],a[maxn]; struct node{ int l,r,cnt; }t[maxn*250]; int root[maxn],q[2][30]; inline void insert(int &x,int l,int r,int pos,int val){ if(!x)x=++tot; t[x].cnt+=val; if(l==r)return ; if(pos<=mid)insert(t[x].l,l,mid,pos,val); else insert(t[x].r,mid+1,r,pos,val); } inline void add(int pos,int val,int d){ while(pos<=n){ insert(root[pos],1,cnt,val,d); pos+=lbt(pos); } } inline int queryt(int l,int r,int k){ if(l==r)return l; int ans=0; for(int i=1;i<=q[0][0];i++)ans-=t[t[q[0][i]].l].cnt; for(int i=1;i<=q[1][0];i++)ans+=t[t[q[1][i]].l].cnt; if(ans>=k){ for(int i=1;i<=q[0][0];i++)q[0][i]=t[q[0][i]].l; for(int i=1;i<=q[1][0];i++)q[1][i]=t[q[1][i]].l; return queryt(l,mid,k); } else{ for(int i=1;i<=q[0][0];i++)q[0][i]=t[q[0][i]].r; for(int i=1;i<=q[1][0];i++)q[1][i]=t[q[1][i]].r; return queryt(mid+1,r,k-ans); } } inline int query(int l,int r,int k){ l--; q[0][0]=q[1][0]=0; while(l){ q[0][++q[0][0]]=root[l]; l-=lbt(l); } while(r){ q[1][++q[1][0]]=root[r]; r-=lbt(r); } return queryt(1,cnt,k); } int main(){ scanf("%d%d",&n,&m); for(int i=1;i<=n;i++)a[i]=read(),hsh[++cnt]=a[i]; char op; for(int i=1;i<=m;i++){ op=getchar(); while(op!='C' &&op!='Q')op=getchar(); if(op=='Q'){ qu[i].l=read(),qu[i].r=read(),qu[i].x=read(); } else if(op=='C'){ qu[i].tp=1; qu[i].l=read(),qu[i].x=read(); hsh[++cnt]=qu[i].x; } } sort(hsh+1,hsh+1+cnt); cnt=unique(hsh+1,hsh+1+cnt)-hsh-1; for(int i=1;i<=n;i++)a[i]=lower_bound(hsh+1,hsh+1+cnt,a[i])-hsh; for(int i=1;i<=m;i++) if(qu[i].tp)qu[i].x=lower_bound(hsh+1,hsh+1+cnt,qu[i].x)-hsh; for(int i=1;i<=n;i++)add(i,a[i],1);//建树 for(int i=1;i<=m;i++){ if(qu[i].tp){ add(qu[i].l,a[qu[i].l],-1); add(qu[i].l,qu[i].x,1); a[qu[i].l]=qu[i].x; } else{ printf("%d\n",hsh[query(qu[i].l,qu[i].r,qu[i].x)]); } } }