这题据dzm大佬的话是并查集板子题,然而卡了我很久,果然还是我太弱了吗……
思路:按秩合并和路径压缩在这道题都能用,对那些用两种优化如同写板子的OIer们算是很友好了。这道题总体思路是把条件相等的合并,条件不等的则判断是否已经在一个集合中,如果是则不成立。具体的来说,离线来做是个不错的选择。先将数据都读入,使e为1的在前,e为0的在后,先全部合并在判断,这样做对最后结果是没有影响的,该矛盾还是要矛盾。这样可以省去记录条件不等的i和j这件麻烦事。因为有些i和j非常大,我们可以在读入的时候模一个大质数(这样其实就可以A掉这道题,但不是最好的办法→离散化)。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define clr(a) memset(a,0,sizeof(a))
using namespace std;
struct haha{LL x,y,e;}a[1000010];
LL n,m,mod=100000007;
LL fa[1000010],h[1000010],uni[1000010<<1];//uni是用来离散化的数组
bool flag;
bool cmp(haha x,haha y){return x.e>y.e;}
LL get(LL x){return fa[x]==x?fa[x]:fa[x]=get(fa[x]);}//路径压缩
void merge(LL x,LL y){//按秩合并
LL fx=get(x);
LL fy=get(y);
if(fx!=fy){
if(h[fx]>=h[fy]){
h[fx]+=h[fy];
fa[fy]=fx;
}
else{
h[fy]+=h[fx];
fa[fx]=fy;
}
}
}
void first(LL k){//初始化
for(LL i=1;i<=k;i++){
h[i]=1;
fa[i]=i;
}
}
int main(){
scanf("%lld",&m);
while(m--){
LL cnt=0;
clr(a);clr(uni);
scanf("%lld",&n);
for(LL i=1;i<=n;i++){
scanf("%lld%lld%lld",&a[i].x,&a[i].y,&a[i].e);
a[i].x%=mod;a[i].y%=mod;//避免过亿的数据,模大质数
uni[++cnt]=a[i].x;uni[++cnt]=a[i].y;
}
sort(uni+1,uni+1+cnt);//45~50:离散化,优化时空效率
LL len=unique(uni+1,uni+1+cnt)-uni-1;
for(LL i=1;i<=n;i++){
a[i].x=lower_bound(uni+1,uni+1+len,a[i].x)-uni-1;
a[i].y=lower_bound(uni+1,uni+1+len,a[i].y)-uni-1;
}
first(len);
flag=0;
sort(a+1,a+1+n,cmp);
for(LL i=1;i<=n;i++){
LL xx=get(a[i].x);
LL yy=get(a[i].y);
if(a[i].e==1)
merge(xx,yy);
if(a[i].e==0)
if(xx==yy){
printf("NO\n");
flag=1;
break;
}
}
if(!flag) printf("YES\n");
}
return 0;
}
功德圆满