[ZJOI2013]K大数查询 Solution

最新推荐文章于 2021-01-05 09:05:38 发布

原创最新推荐文章于 2021-01-05 09:05:38 发布 · 149 阅读

0 ·

CC 4.0 BY-SA版权

Solution 专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种处理集合操作和查询的高效算法。通过离线所有操作并使用整体二分法，算法能在O(nlog^2n)的时间复杂度内解决在多个集合上进行元素添加和查询的问题。关键步骤包括将操作按类型分类，对添加操作用树状数组维护，对查询操作计算区间值，最后通过递归左右半部分确定答案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给 $n$ 个集合，最开始都为空集。
有 $k$ 个操作，操作 $1$ ：每次在 $l - r$ 这些集合里面都加入一个数；操作 $2$ ：查询 $l - r$ 这些集合的并集中的 $x$ 大的值。
考虑整体二分。
把所有操作都离线下来。
对于二分的答案 $m i d$ ，检索所有的操作：

添加操作：如果这次加入的值大于 $m i d$ ，对应区间加 $1$ ，然后加入递归右半部分，如果不大于 $m i d$ ，那么直接加入递归左半部分。
查询操作：查询一下对应区间的区间和 $v a l$ ，如果大于这个询问的排名 $k t h$ ，就加入递归右半部分，反之让 $k t h$ 减去 $v a l$ ，加入左半部分，这是为了保证时间复杂度。

$c o d e :$

/*考虑那些修改产生的影响：
先考虑添加操作，如果值大于mid，那么树状数组维护区间加。 
然后是查询操作，算一下区间大于这个数的个数，如果多了，放入右边，否则放入左边 
*/
#include <bits/stdc++.h>
#define int long long
#define regi register long long
int n,m;
int cnt;
int c[2][60000];
int ans[51000];
struct operate{
	int l;
	int r;
	int id;
	int val;
	int type;
}d[51000],tmpl[51000],tmpr[51000];
int read(){
	int r=0,w=0,c;
	for(;!isdigit(c=getchar());r=c);
	for(w=c^48;isdigit(c=getchar());w=w*10+(c^48));
	return r^45?w:-w;
}
void add(int op,int x,int val){
	for(x;x<=n;x+=x&-x)
	  c[op][x]+=val;
}
void ADD(int x,int y,int val){
	add(0,x,val);
	add(0,y+1,-val);
	add(1,x,x*val);
	add(1,y+1,-(y+1)*val);
}
int ask(int op,int x){
	int s=0;
	for(x;x>0;x-=x&-x)
	  s+=c[op][x];
	return s;
}
int ASK(int x,int y){
  return (y+1)*ask(0,y)-ask(1,y)-x*ask(0,x-1)+ask(1,x-1);
}
void solve(int l,int r,int ql,int qr){
	if(ql>qr)
	  return;
	if(l==r){
		for(regi i=ql;i<=qr;++i){
		  if(d[i].type==2){
		    ans[d[i].id]=l;
		  }
		}
		return;
	}
	int mid=l+r>>1;
	int tl=0,tr=0;
	for(regi i=ql;i<=qr;++i){
		if(d[i].type==1){
			if(d[i].val>mid){
			  ADD(d[i].l,d[i].r,1);
			  tmpr[++tr]=d[i];
			}
			else
			  tmpl[++tl]=d[i];
		}
		else{
			int res=ASK(d[i].l,d[i].r);
			if(res>=d[i].val)
			  tmpr[++tr]=d[i];
			else{
				d[i].val-=res;
			  tmpl[++tl]=d[i];
			}
		}	
	}
	for(regi i=1;i<=tr;++i)
	  if(tmpr[i].type==1&&tmpr[i].val>mid)
	    ADD(tmpr[i].l,tmpr[i].r,-1);
	for(regi i=1;i<=tr;++i)
	  d[ql+i-1]=tmpr[i];
	for(regi i=1;i<=tl;++i)
	  d[ql+tr+i-1]=tmpl[i];
  solve(mid+1,r,ql,ql+tr-1);
	solve(l,mid,ql+tr,qr);
}
main(){
	n=read();
	m=read();
	for(regi i=1;i<=m;++i){
	  d[i].type=read();
	  d[i].l=read();
	  d[i].r=read();
	  d[i].val=read();
	  d[i].id=(d[i].type==2?++cnt:0);
	}
	solve((-1LL)<<63,(1LL<<63)-1,1,m);
	for(regi i=1;i<=cnt;++i)
	  printf("%lld\n",ans[i]);
	return 0;
}