2023牛客暑期多校训练营5-E Red and Blue and Green

2023牛客暑期多校训练营5-E Red and Blue and Green

https://ac.nowcoder.com/acm/contest/57359/E

题意

解题思路

由题意可见对于任意两个区间，要么相交，要么包含。我们可以将区间建成一棵树，有直接包含关系的区间是父子，有同一个父亲的区间是兄弟。

方便起见，额外设置区间$[i,i](1\le i\le n)$，显然$w$为$0$（可知出现$[i,i,1]$时排列不存在）。先设 p = { 1 , 2 , 3 , 4 , 5... } p=\{1,2,3,4,5...\} p={1,2,3,4,5...}。

对于一个区间，假设其子区间已经合法，如果相加使本身也合法，则无需操作；若相加不能使本身合法，则交换任意相邻两个区间的最大值与最小值，可以保证其逆序数对加一且不影响子区间逆序数对数。

注意维护每一区间的最大值与最小值。

代码

#include<bits/stdc++.h>
using namespace std;
int n,m,p[5003],fa[5003],vis[5003],pr[5003],re[5003];
vector<int>ve[5003];
struct node{
	int l,r,w;
}a[5003];
bool cmp(node a,node b){
	if(a.l!=b.l)return a.l<b.l;
	return a.r>b.r;
}
void dfs(int u){
	int c=0,sum=0;
	for(int i=u+1;i<=m;i++){
		if(a[i].l>a[u].r)break;
		if(a[i].l<=c)continue;
		ve[u].push_back(i);
		fa[i]=u;
		sum+=a[i].w;
		c=a[i].r;
		dfs(i);
	}
	if(ve[u].size()==0){
		pr[u]=a[u].l,re[u]=a[u].r;
		return;
	}
	if(sum%2!=a[u].w){
		swap(p[re[ve[u][0]]],p[pr[ve[u][1]]]);
	}
	int mi=0x3f3f3f3f,Mi=0,ma=0,Ma=0;
	for(int i=a[u].l;i<=a[u].r;i++){
		if(p[i]<mi){
			mi=p[i];
			Mi=i;
		}
		if(p[i]>ma){
			ma=p[i];
			Ma=i;
		}
	}
	pr[u]=Mi,re[u]=Ma;
}
int main(){
	int flag=1;
	cin>>n>>m;
	for(int i=1;i<=n;i++)p[i]=i;
	for(int i=1;i<=m;i++){
		cin>>a[i].l>>a[i].r>>a[i].w;
		if(a[i].l==a[i].r&&a[i].w==1){
			flag=0;
		}
	}
	if(!flag){
		cout<<-1;
		return 0;
	}
	for(int i=1;i<=n;i++)a[i+m].l=i,a[i+m].r=i,a[i+m].w=0;
	m+=n;
	sort(a+1,a+m+1,cmp);
	int b=0;
	for(int i=1;i<=m;i++){
		if(a[i].l>b){
			b=a[i].r,dfs(i);
		} 
	}
	for(int i=1;i<=n;i++)cout<<p[i]<<' ';
}