蓝桥杯 双向排序（线段树）

3419. 双向排序 - AcWing

---

规定pre_q为升序区间的第一个数，整个区间左半侧为降序区间，右半侧为升序区间

当处理操作时，如果令1~q降序，则利用pre_q判断1~q是否已经降序，升序同理

区间修改时只需要传入需要修改的位置个数k，从根节点开始往下找，直到将不为目标状态的k个位置更新为目标状态为止，然后更新pre_q

注意懒标记的使用

#include<iostream>
#include<vector>
using namespace std;
const int N=1e5+5;
typedef long long ll; 
struct node
{
	int l,r;
	int sum;//1的个数 
	int lazy;
}tree[N*4];
vector<int> res[2];
void pushup(int u)
{
	tree[u].sum=tree[u<<1].sum+tree[u<<1|1].sum;
}
void build(int u,int l,int r)
{
	if(l==r) tree[u]={l,r,1,-1};
	else
	{
		tree[u].l=l;
		tree[u].r=r;
		tree[u].lazy=-1;
		int mid=l+r>>1;
		build(u<<1,l,mid);
		build(u<<1|1,mid+1,r);
		pushup(u);
	}
}
void pushdown(int u)
{
	if(tree[u].lazy==-1) return;
	int lazy=tree[u].lazy;
	tree[u<<1].lazy=lazy;
	tree[u<<1].sum=(tree[u<<1].r-tree[u<<1].l+1)*lazy;
	tree[u<<1|1].lazy=lazy;
	tree[u<<1|1].sum=(tree[u<<1|1].r-tree[u<<1|1].l+1)*lazy;
	tree[u].lazy=-1;
}
void modify0(int u,int cnt)
{
	int cnt_1=tree[u].sum;
	if(cnt_1<=cnt)
	{
		tree[u].sum=0;
		tree[u].lazy=0;
		return;
	}
	pushdown(u);
	int cnt_l1=tree[u<<1].sum;
	int cnt_r1=tree[u<<1|1].sum;
	if(cnt_l1>=cnt) modify0(u<<1,cnt);
	else
	{
		modify0(u<<1,cnt_l1);
		modify0(u<<1|1,cnt-cnt_l1);
	}
	pushup(u);
}
void modify1(int u,int cnt)
{
	int cnt_0=(tree[u].r-tree[u].l+1)-tree[u].sum;
	if(cnt_0<=cnt)
	{
		tree[u].sum=tree[u].r-tree[u].l+1;
		tree[u].lazy=1;
		return;
	}
	pushdown(u);
	int cnt_l0=(tree[u<<1].r-tree[u<<1].l+1)-tree[u<<1].sum;
	int cnt_r0=(tree[u<<1|1].r-tree[u<<1|1].l+1)-tree[u<<1|1].sum;
	if(cnt_l0>=cnt) modify1(u<<1,cnt);
	else
	{
		modify1(u<<1,cnt_l0);
		modify1(u<<1|1,cnt-cnt_l0);
	}
	pushup(u);
}
int query(int u,int pos)
{
	if(tree[u].l==tree[u].r) return tree[u].sum;
	pushdown(u);
	int mid=tree[u].l+tree[u].r>>1;
	if(pos<=mid) return query(u<<1,pos);
	else return query(u<<1|1,pos);
}
int main()
{
	int n,m;cin>>n>>m;
	build(1,1,n);
	int pre_q=1;//分界点,规定为升序区间的第一个数 
	while(m--)
	{
		int p,q;cin>>p>>q;
		if(!p)//降 
		{
			if(q<pre_q) continue;
			modify0(1,q-pre_q+1);
			pre_q=q+1;
		}
		else//升 
		{
			if(q>=pre_q) continue;
			modify1(1,pre_q-q);
			pre_q=q;
		}
	}
	for(int i=1;i<=n;i++)
		if(query(1,i))
			res[1].push_back(i);
		else 
			res[0].push_back(i);
	for(int i=res[0].size()-1;i>=0;i--)
		cout<<res[0][i]<<" ";
	for(int i=0;i<res[1].size();i++)
	{
		cout<<res[1][i];
		if(i!=res[1].size()-1) cout<<" ";
	}
}