Dynamic Rankings

本来想用这个题来写一下整体二分。
不过看了很久的树套树神奇数据结构貌似还从来没写过。
来练下手好了。

考虑我们没有修改操作时,区间\(kth\)可以用主席树来做,相当于维护了一个前缀和(这个前缀和的形式是由权值线段树来表达的)。
但是我们有修改操作了,如果我们继续维护这个暴力的前缀和,那么每次修改是\(O(n)\)的。
但是如果我们套上一个树状数组,考虑变成了单点加到区间查询就变得非常容易,这样的代价是我们从\(O(log)\)的一次查询,变成每次要同时跑\(log\)颗树。
是一个\(O(nlog^2n)\)的算法。

Dynamic Rankings
#include<iostream>
#include<algorithm>
#include<cstdio>
#define ll long long
#define N 200005
#define lowbit(x) (x & -x)

ll n,m;

ll a[N],num[N << 1];//数集 

struct P{
  int opt;
  ll l,r,k;
}e[N];

ll head[N];//树状数组对应的线段树的头结点。 

struct E{ll ls,rs,val;}T[N * 400]; 

#define l(x) T[x].ls
#define r(x) T[x].rs
#define v(x) T[x].val 
#define mid ((l + r) >> 1) 

ll cnt;

inline void add(ll &now,int l,int r,int to,int ki){
  if(!now)now = ++cnt;
//	std::cout<<now<<" "<<l<<" "<<r<<" "<<to<<" "<<ki<<std::endl; 	
  v(now) += ki;
  if(l == r)return;
  if(to <= mid)
  add(l(now),l,mid,to,ki);
  else
  add(r(now),mid + 1,r,to,ki);
}

inline void change(int now,int val){
  ll k = std::lower_bound(num + 1,num + num[0] + 1,a[now]) - num;
  for(int i = now;i <= num[0];i += lowbit(i))
  add(head[i],1,num[0],k,val);
}

ll t1[N << 2],t2[N << 2];

inline ll find(int l,int r,int k){
//	std::cout<<l<<" "<<r<<" "<<k<<std::endl;
  if(l == r)return l;
  ll sum = 0;
  for(int i = 1;i <= t2[0];++i)
  sum += v(l(t2[i]));
  for(int i = 1;i <= t1[0];++i)
  sum -= v(l(t1[i]));
//	std::cout<<sum<<std::endl;
  if(sum >= k){
  	for(int i = 1;i <= t1[0];++i)
  	t1[i] = l(t1[i]);
  	for(int i = 1;i <= t2[0];++i)
  	t2[i] = l(t2[i]);
  	return find(l,mid,k);
  }
  else{
  	for(int i = 1;i <= t1[0];++i)
  	t1[i] = r(t1[i]); 
  	for(int i = 1;i <= t2[0];++i)
  	t2[i] = r(t2[i]);
  	return find(mid + 1,r,k - sum);
  }
}

inline ll pre(ll l,ll r,ll k){
  t1[0] = 0;
  t2[0] = 0;
  for(int i = l - 1;i;i -= lowbit(i))
  t1[++t1[0]] = head[i];
  for(int i = r;i;i -= lowbit(i))
  t2[++t2[0]] = head[i];
//	for(int i = 1;i <= t1[0];++i)
//	std::cout<<t1[i]<<" ";
//	puts("");
//	for(int i = 1;i <= t2[0];++i)
//	std::cout<<t2[i]<<" "; 
//	puts("");
  return find(1,num[0],k);
} 

int main(){
  scanf("%lld%lld",&n,&m);
  for(int i = 1;i <= n;++i)
  scanf("%lld",&a[i]),num[++num[0]] = a[i];
  for(int i = 1;i <= m;++i){
  	char a;
  	while(a != 'Q' && a != 'C')
  	a = getchar();
  	if(a == 'Q'){
  		e[i].opt = 1;
  		scanf("%lld%lld%lld",&e[i].l,&e[i].r,&e[i].k);
  	}else{
  		e[i].opt = 2;
  		scanf("%lld%lld",&e[i].k,&e[i].l);
  		num[++num[0]] = e[i].l;
  	}
  	a = 'z';
  }
  std::sort(num + 1,num + num[0] + 1);
  num[0] = std::unique(num + 1,num + num[0] + 1) - num - 1;
  for(int i = 1;i <= n;++i)
  change(i,1);
//	for(int i = 1;i <= n;++i)
//	std::cout<<head[i]<<std::endl;
  for(int i = 1;i <= m;++i){
  	if(e[i].opt == 1){
  		std::cout<<num[pre(e[i].l,e[i].r,e[i].k)]<<std::endl;
  	}else{
  		change(e[i].k,-1);
  		a[e[i].k] = e[i].l;
  		change(e[i].k,1);
  	} 
  }
}
posted @ 2021-05-17 12:07  fhq_treap  阅读(84)  评论(0编辑  收藏  举报