Dynamic Rankings
动态区间第K大,树状数组套主席树。
树套树和整体二分当然也能做啊。
之前我们写主席树是前缀和,但是可以快速修改和区间求和的就容易想到树状数组,于是就树状数组,每个bit存一个主席树的根就行了。
UPD:人弱常数大,开O2+快读才过了luogu上新加的10个点...
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=100005;
int rt[N<<9],ls[N<<9],rs[N<<9],siz[N<<9],tot,a[N],n,m,tmp,lsh[N<<3],ci[N],cj[N],ck[N],u,totx,toty,tx[N],ty[N];
void update(int &k,int pre,int l,int r,int pos,int val) {
ls[k=++tot]=ls[pre],rs[k]=rs[pre],siz[k]=siz[pre]+val;
if(l==r) return;
int mid=l+r>>1;
if(pos<=mid) update(ls[k],ls[pre],l,mid,pos,val);
else update(rs[k],rs[pre],mid+1,r,pos,val);
}
int query(int l,int r,int k) {
if(l==r) return l;
int mid=l+r>>1;
int sum=0;
for(int i=1;i<=totx;i++) sum-=siz[ls[tx[i]]];
for(int i=1;i<=toty;i++) sum+=siz[ls[ty[i]]];
if(k<=sum) {
for(int i=1;i<=totx;i++) tx[i]=ls[tx[i]];for(int i=1;i<=toty;i++) ty[i]=ls[ty[i]];
return query(l,mid,k);
}
else {
for(int i=1;i<=totx;i++) tx[i]=rs[tx[i]];for(int i=1;i<=toty;i++) ty[i]=rs[ty[i]];
return query(mid+1,r,k-sum);
}
}
void add(int x,int v) {
int k=lower_bound(lsh+1,lsh+1+u,a[x])-lsh;
for(int i=x;i<=n;i+=i&-i) update(rt[i],rt[i],1,u,k,v);
}
int main() {
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]),lsh[++tmp]=a[i];
static char s[5];
for(int i=1;i<=m;i++) {
scanf("%s",s);
if(s[0]=='Q')
scanf("%d%d%d",&ci[i],&cj[i],&ck[i]);
else scanf("%d%d",&ci[i],&cj[i]),lsh[++tmp]=cj[i];
}
sort(lsh+1,lsh+1+tmp);u=unique(lsh+1,lsh+1+tmp)-lsh-1;
for(int i=1;i<=n;i++) add(i,1);
for(int i=1;i<=m;i++) {
if(ck[i]) {
totx=toty=0;
for(int j=ci[i]-1;j;j-=j&-j) tx[++totx]=rt[j];
for(int j=cj[i];j;j-=j&-j) ty[++toty]=rt[j];
printf("%d\n",lsh[query(1,u,ck[i])]);
}
else add(ci[i],-1),a[ci[i]]=cj[i],add(ci[i],1);
}
}
我是咸鱼。转载博客请征得博主同意Orz