[HEOI2016/TJOI2016]排序-题解

• 【题目地址】

【本题有O(nlogn)的线段树分裂合并的做法】但是这里由于这个方法实现较复杂我才不会告诉你我不会，所以这里讲另外一种十分巧妙的技巧。

---

我们来看，如果每次对于每个操作直接排序的话，复杂度会达到$O(n^{2}logn)$，而题目只在最后询问一次，所以我们考虑将操作存下来，离线进行计算。

考虑如何进行快速的对一段区间进行排序：

普通的为$O(nlogn)$肯定承受不了，但是我们只需要知道排序后的第$q$个数是多少，所以我们只用考虑将比答案小的排前面，大的排后面即可，所以我们可以选取一个值作为答案，显然这个值是有单调性的，所以我们可以将原序列全部排序后在上面二分这个值，而对于二分了的这个值，我们把原序列变成一个01序列，0表示比这个二分值小，1表示大。

那么继而考虑如何对于一段01序列排序，其实我们只需知道这一段的01个数，然后如果是升序，就将原来序列的前面0的那么多个数字变成0，后面的全部变成1即可；降序就是反起来。

那么用线段树区间修改查询，我们可以做到每次$logn$的排序，所以每次二分了一个值后，我们就将序列转换，进行操作，最后判断第$q$个位置上是0还是1，如果是1，那么就表示答案还要大一些，否则就小一些。

总的复杂度加上二分，就是$O(nlo{g}^{2}n)$，但是不能应对强制在线和多组询问。

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int M=1e5+10;

int n,m,Q,val[M],bit[M];
int cnt[M<<2],lazy[M<<2];

void pushup(int o){
	cnt[o]=cnt[o<<1]+cnt[o<<1|1];
}
void pushdown(int o,int l,int r,int mid){
	if(lazy[o]==-1) return;
	lazy[o<<1]=lazy[o<<1|1]=lazy[o];
	if(!lazy[o]){
		cnt[o<<1]=cnt[o<<1|1]=0;
	}else{
		cnt[o<<1]=(mid-l+1);
		cnt[o<<1|1]=(r-mid);
	}
	lazy[o]=-1;
}

void build(int o,int l,int r){
	lazy[o]=-1;
	if(l==r){
		cnt[o]=(bit[l]==1);
		return;
	}
	int mid=l+r>>1;
	build(o<<1,l,mid);
	build(o<<1|1,mid+1,r);
	pushup(o);
}
int Find_Bit(int o,int l,int r,int L,int R){
	if(L<=l&&r<=R) return cnt[o];
	if(!cnt[o]) return 0;
	int mid=l+r>>1;
	pushdown(o,l,r,mid);
	if(R<=mid) return Find_Bit(o<<1,l,mid,L,R);
	else if(L>mid) return Find_Bit(o<<1|1,mid+1,r,L,R);
	else return Find_Bit(o<<1,l,mid,L,R)+Find_Bit(o<<1|1,mid+1,r,L,R);
}
void Update(int o,int l,int r,int L,int R,int v){
	if(L<=l&&r<=R){
		cnt[o]=(v==1)*(r-l+1);
		lazy[o]=v;
		return;
	}
	int mid=l+r>>1;
	pushdown(o,l,r,mid);
	if(L<=mid) Update(o<<1,l,mid,L,R,v);
	if(R>mid) Update(o<<1|1,mid+1,r,L,R,v);
	pushup(o);
}
int Query(int o,int l,int r,int p){
	if(l==r) return cnt[o];
	int mid=l+r>>1;
	pushdown(o,l,r,mid);
	if(!cnt[o]) return 0;
	if(p<=mid) return Query(o<<1,l,mid,p);
	else return Query(o<<1|1,mid+1,r,p);
}
struct Opt{
	int opt,l,r;
	void in(){scanf("%d%d%d",&opt,&l,&r);}
	Opt(){}
	Opt(int a,int b,int c):opt(a),l(b),r(c){}
}Op[M];
bool solve(int mv){
	for(int i=1;i<=n;i++){
		if(val[i]<=mv)bit[i]=0;
		else bit[i]=1;
	}
	build(1,1,n);
	for(int i=1,num;i<=m;i++){
		num=Find_Bit(1,1,n,Op[i].l,Op[i].r);
		if(Op[i].opt){
			Update(1,1,n,Op[i].l,Op[i].l+num-1,1);
			Update(1,1,n,Op[i].l+num,Op[i].r,0);
		}else{
			num=Op[i].r-Op[i].l+1-num;
			Update(1,1,n,Op[i].l,Op[i].l+num-1,0);
			Update(1,1,n,Op[i].l+num,Op[i].r,1);
		}
	}
	return Query(1,1,n,Q);
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&val[i]);
	for(int i=1;i<=m;i++)Op[i].in();
	scanf("%d",&Q);
	int l=1,r=n,mid;
	while(l<r){
		mid=l+r>>1;
		if(solve(mid)){
			l=mid+1;
		}else{
			r=mid;
		}
	}
	printf("%d\n",l);
	return 0;
}

---

这里有个技巧就是对原序列排序时间复杂度过高，但对一个01序列排序则可以大大降低时间复杂度。【[APIO2018]的二分专题讲解的课件上还有一些关于二分01序列排序的技巧】