2023牛客暑期多校训练营5-E Red and Blue and Green
题目描述
思路
通过描述可知,区间不会存在 只重叠一部分 的现象,即只有 完全包含 和 互不相交 两种情况。这和树很类似,所以选择用树来维护区间内的关系。
我们可以通过排序区间 左端点 再 右端点 来表明父亲与孩子的关系。
接下来是解决逆序数对的问题。 首先如果一个区间内逆序数对为奇数个,则将区间最后的两个数交换即可;若为偶数则无须改变。 当子区间的逆序数对成功排列后,考虑父亲区间。若子区间逆序数对个数奇偶性与父亲区间相同,则无须变更,不同,我们只需要将任意两个区间中左区间最大值与右区间最小值交换即可。
如图,在讨论父亲区间时,只需要将子区间的左最大与右最小交换即可,及a[i].r与a[i+1].l(此处是边界,点位置会发生变化,可以用pos[i]记录i的位置变化)
- 同时可能会问:如果子区间不足两个该怎么办?
因为这里只存在完全包含或互不相交的包含和并列关系,所以我们可以添加[i,i]区间以保证每个区间长度大于一的区间都有子区间也方便更换。
代码如下
#include <bits/stdc++.h>
using namespace std;
int n,m;
const int N=2e3+2;
struct ppap{
int l,r,w;
}a[N];
int cmp(ppap x,ppap y)//区间排序
{
if(x.l!=y.l) return x.l<y.l;
return x.r>y.r;
}
int fa[N],so[N];
vector<int> tr[N];
int pos[N],gr[N];//pos是点的位置,实时更新
int cnt=1;//从1开始可以留0给总区间
void trp()//建树
{
fa[1]=0;//最大的统一祖先
tr[0].push_back(1);
for(int i=2;i<cnt;i++)
{
if(a[i].r<=a[i-1].r){//如果是包含关系
tr[i-1].push_back(i);
fa[i]=i-1;
}
else if(a[i].l>a[i-1].r){//如果是并列关系
int x=fa[i-1];
while(a[x].r<a[i].r)//向上找父亲,直到找到一个能包含的区间
x=fa[x];
tr[x].push_back(i);
fa[i]=x;
}
}
}
void dfs(int x)
{
int q=a[x].w;
int sum=0;
for(auto i:tr[x])
{
if(i==fa[x]) continue ;
dfs(i);
sum+=a[i].w;//记录所有子区间的逆序数对奇偶性
}
if((sum%2)==q) return ;//如果刚好相同则不用更改
else if(x!=0){//这里写复杂了,只需找到左区间的最大与右区间的最小
int a1=0,a2=0,a3=0,a4=0;
for(auto i:tr[x])
{
if(i==fa[x]) continue ;
if(a3==0&&a1!=0) a3=a[i].l,a4=a[i].r;
if(a1==0) a1=a[i].l,a2=a[i].r;
}
int k1,k2;
if(a1>a4) k1=a1,k2=a4;
else k1=a3,k2=a2;
swap(gr[pos[k1]],gr[pos[k2]]);//更换数组中的量
int pp=pos[k1];
pos[k1]=pos[k2];
pos[k2]=pp;
}
}
int main()
{
scanf("%d%d",&n,&m);
int flag=0;
a[0].l=1;
a[0].r=n;
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&a[cnt].l,&a[cnt].r,&a[cnt].w);
if(a[cnt].l==a[cnt].r&&a[cnt].w==1){//判断是否存在不合理的情况
flag=1;
}
if(a[cnt].l!=a[cnt].r) cnt++;//只读入长度大于1的区间防止之后添加[i,i]时重复
}
for(int i=1;i<=n;i++)//添加[i,i]
{
a[cnt].l=a[cnt].r=i;
a[cnt].w=0;
cnt++;
}
if(flag) {
puts("-1");
return 0;
}
sort(a+1,a+cnt,cmp);//将区间排序
trp(); //建树
for(int i=1;i<=n;i++) {
gr[i]=i;
pos[i]=i;
}
dfs(0);//从统一祖先开始可以遍历到所有区间
for(int i=1;i<=n;i++) printf("%d ",gr[i]);
}