正题
评测记录:https://www.luogu.org/recordnew/lists?uid=52918&pid=P1955
题目大意
给出若干个关系,xi≠xj or xi=xjxi≠xjorxi=xj,然后求是否全部都满足。
解题思路
首先由于数据较大,所以我们先离散化,然后用并查集。先处理所有等于的关系,然后不等于的再判断。
code
#include<cstdio>
#include<algorithm>
#define N 1000010
using namespace std;
struct node{int x,y,t;
}a[N];
int n,uniq[2*N],f[2*N],t;
bool cmp(node x,node y)
{return x.t>y.t;
}
int find(int x)
{if(f[x]!=x) return f[x]=find(f[x]);return x;
}
int main()
{scanf("%d",&t);for(int ti=1;ti<=t;ti++){scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].t);uniq[i]=a[i].x;uniq[i+n]=a[i].y;}//以上为去重stable_sort(uniq+1,uniq+1+2*n);int m=unique(uniq+1,uniq+1+2*n)-(uniq+1);//以上为离散化for(int i=1;i<=m;i++) f[i]=i;//初始化并查集for(int i=1;i<=n;i++){a[i].x=lower_bound(uniq+1,uniq+1+m,a[i].x)-(uniq+1),a[i].y=lower_bound(uniq+1,uniq+1+m,a[i].y)-(uniq+1);}//以上为找到离散化之后的位置stable_sort(a+1,a+1+n,cmp);bool flag=0;for(int i=1;i<=n&&!flag;i++){int fa=find(a[i].x),fb=find(a[i].y);if(a[i].t) f[min(fa,fb)]=max(fa,fb);//链接else if(fa==fb) printf("NO\n"),flag=true;//判答案}if(!flag) printf("YES\n");}
}