动态区间第k小:树状数组套权值线段树

所谓树状数组套权值线段树,就是在树状树组上套权值线段树

(逃)

解析

如何解决静态区间第k小?
使用主席树就ok啦

辣么如何解决动态区间第k小嘞…

我们想想主席树为啥不能解决动态区间第k小
因为如果改了一个点的值,后面所以的权值线段树都需要修改
单次修改的时空复杂度为 n l o g n nlogn nlogn,无法承受

仔细想想,静态的主席树似乎就是一个关于值域的高级一点的前缀和
暴力修改前缀和当然是需要修改 O ( n ) O(n) O(n)个了
但是既然是前缀和,我们为就可以想到用树状数组优化
这样需要修改的东西就变成了 O ( l o g n ) O(logn) O(logn)个了

具体的说,和树状数组的定义类似,第i棵权值线段树维护的是 [ i − l o w b i t ( i ) + 1 , i ] [i-lowbit(i)+1,i] [ilowbit(i)+1,i]的值域
这样单次修改只需要改log棵树
时空复杂度 n l o g n 2 nlogn^2 nlogn2
询问的时候把对应的log棵树都拿出来
一起算size就可以了
时间复杂度 n l o g n 2 nlogn^2 nlogn2
可以说是很优秀了
板子传送门

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+100;
inline ll read(){
	ll x=0,f=1;char c=getchar();
	while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
	while(isdigit(c)){x=x*10+c-'0';c=getchar();}
	return x*f;
}
int n,m;
int r[N],tot;
struct node{
	int ls,rs,siz;
}tr[N*300];
inline int New(){
	++tot;tr[tot].ls=tr[tot].rs=0;
	return tot;
}
#define mid ((l+r)>>1)
inline void pushup(int k){
	if(k) tr[k].siz=tr[tr[k].ls].siz+tr[tr[k].rs].siz;return;
}
inline void upd(int &k,int l,int r,int p,int v){
	if(!k) k=New();
	if(l==r){
		tr[k].siz+=v;return;
	}
	if(p<=mid) upd(tr[k].ls,l,mid,p,v);
	else upd(tr[k].rs,mid+1,r,p,v);
	pushup(k);
	//printf("  k=%d (%d %d),siz=%d\n",k,l,r,tr[k].siz);
}
int a[N],q[N],cnt;
inline void change(int p,int v){
	for(int i=p;i<=n;i+=i&-i){
		upd(r[i],1,cnt,a[p],-1);upd(r[i],1,cnt,v,1);
	}
	a[p]=v;
	return;
}
vector<int>L,R;
void get(int l,int rr){
	L.clear();R.clear();
	for(int i=l-1;i;i-=i&-i) L.push_back(r[i]);
	for(int i=rr;i;i-=i&-i) R.push_back(r[i]);
	return;
}
int ask(int l,int r,int k){
	if(l==r) return q[l];
	int num=0;
	for(int i=0;i<R.size();i++){
		int x=R[i];num+=tr[tr[x].ls].siz;
	}
	for(int i=0;i<L.size();i++){
		int x=L[i];num-=tr[tr[x].ls].siz;
	}
	if(k<=num){
		for(int i=0;i<R.size();i++){
			R[i]=tr[R[i]].ls;
		}
		for(int i=0;i<L.size();i++){
			L[i]=tr[L[i]].ls;
		}
		return ask(l,mid,k);
	}
	else{
		for(int i=0;i<R.size();i++){
			R[i]=tr[R[i]].rs;
		}
		for(int i=0;i<L.size();i++){
			L[i]=tr[L[i]].rs;
		}
		return ask(mid+1,r,k-num);
	}
}
char op[N];
int x[N],y[N],k[N];
int main(){
	n=read();m=read();
	for(int i=1;i<=n;i++){
		a[i]=read();q[++cnt]=a[i];
	}
	for(int i=1;i<=m;i++){
		scanf(" %c",&op[i]);x[i]=read();y[i]=read();
		if(op[i]=='Q') k[i]=read();
		else q[++cnt]=y[i];
	}
	sort(q+1,q+1+cnt);
	cnt=unique(q+1,q+1+cnt)-q-1;
	for(int i=1;i<=n;i++){
		//printf("\ni=%d\n",i);
		a[i]=lower_bound(q+1,q+1+cnt,a[i])-q;
		for(int p=i;p<=n;p+=p&-p){
			//printf("upd:%d\n",p);
			upd(r[p],1,cnt,a[i],1);
		}
	}
	//for(int i=1;i<=n;i++) printf("i=%d rot=%d siz=%d\n",i,r[i],tr[r[i]].siz);
	for(int i=1;i<=m;i++){
		if(op[i]=='Q'){
			get(x[i],y[i]);
			printf("%d\n",ask(1,cnt,k[i]));
		}
		else{
			y[i]=lower_bound(q+1,q+1+cnt,y[i])-q;
			change(x[i],y[i]);
		}
	}
	return 0;
}

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值