HDU - 5649 DZY Loves Sorting 线段树+二分 转化为0 1

题目链接:点击查看

题意:n个数,m次操作,0 l r,区间排序成递增,1 l r,区间排序成递减,最后输出k位置的数

题解:二分答案,每次大于等于这个数的设为1,小于的设为0,最后查看k位置的这个数是否为1,为1即符合

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
struct node{
	int l,r;
	int val[2];
	int laz;
}tree[N<<2];
struct node1{
	int op,l,r;
}a[N];
int n,m,k,b[N],mi;
void pushup(int cur)
{
	tree[cur].val[1]=tree[cur<<1].val[1]+tree[cur<<1|1].val[1];
	tree[cur].val[0]=tree[cur<<1].val[0]+tree[cur<<1|1].val[0];
}
void build(int l,int r,int cur)
{
	tree[cur].l=l;
	tree[cur].r=r;
	tree[cur].laz=-1;
	if(l==r)
	{
		tree[cur].val[1]=b[l]>=mi;
		tree[cur].val[0]=b[l]<mi;
		return;
	}
	int mid=(r+l)>>1;
	build(l,mid,cur<<1);
	build(mid+1,r,cur<<1|1);
	pushup(cur);
}
int num0,num1;
void pushdown(int cur)
{
	if(tree[cur].laz!=-1)
	{
		tree[cur<<1].laz=tree[cur].laz;
		tree[cur<<1].val[tree[cur].laz]=tree[cur<<1].r-tree[cur<<1].l+1;
		tree[cur<<1].val[tree[cur].laz^1]=0;
		tree[cur<<1|1].laz=tree[cur].laz;
		tree[cur<<1|1].val[tree[cur].laz]=tree[cur<<1|1].r-tree[cur<<1|1].l+1;
		tree[cur<<1|1].val[tree[cur].laz^1]=0;
		tree[cur].laz=-1;
	}
}
void update(int pl,int pr,int cur,int val)
{
	if(pl<=tree[cur].l&&tree[cur].r<=pr)
	{
		tree[cur].val[val]=tree[cur].r-tree[cur].l+1;
		tree[cur].val[val^1]=0;
		tree[cur].laz=val;
		return;
	}
	pushdown(cur);
	if(pl<=tree[cur<<1].r) update(pl,pr,cur<<1,val);
	if(pr>=tree[cur<<1|1].l) update(pl,pr,cur<<1|1,val);
	pushup(cur); 
}
void query(int pl,int pr,int cur)
{
	if(pl<=tree[cur].l&&tree[cur].r<=pr)
	{
		num0+=tree[cur].val[0];
		num1+=tree[cur].val[1];
		return;
	}
	pushdown(cur);
	if(pl<=tree[cur<<1].r) query(pl,pr,cur<<1);
	if(pr>=tree[cur<<1|1].l) query(pl,pr,cur<<1|1);
	pushup(cur);
}
int judge(int x)
{
	build(1,n,1);
	int l,r;
	for(int i=1;i<=m;i++)
	{
		l=a[i].l;r=a[i].r;
		num0=0;num1=0;
		query(l,r,1);
		if(a[i].op)
		{
			if(num1) update(l,l+num1-1,1,1);
			if(num0) update(r-num0+1,r,1,0);
		}
		else
		{
			if(num0) update(l,l+num0-1,1,0);
			if(num1) update(r-num1+1,r,1,1);
		}
	} 
	num1=0;
	query(k,k,1);
	return num1;
}
int main()
{
	int T;scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)scanf("%d",&b[i]);
		for(int i=1;i<=m;i++)scanf("%d%d%d",&a[i].op,&a[i].l,&a[i].r);
		scanf("%d",&k);
		int l=1,r=n;
		int ans;
		while(l<=r)
		{
			mi=(r+l)>>1;
			if(judge(mi)) ans=mi,l=mi+1;
			else r=mi-1;
		}
		printf("%d\n",ans);
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/mmk27_word/article/details/89005462