解题思路
先把所有相等情况加进并查集,在判不等就行了。
代码:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
#include<cmath>
#include<cstdlib>
using namespace std;
struct ldx{
int x,y,c;
bool operator < (const ldx p) const{
return c>p.c;
}
}s[100005];
int b[200005];
int fa[200005];
int n,tot,T;
int zh(int x){
if(x==fa[x]) return x;
else return fa[x]=zh(fa[x]);
}
int main(){
scanf("%d",&T);
while(T--){
scanf("%d",&n);tot=0;
for(int i=1;i<=n*2;i++) fa[i]=i;
for(int i=1;i<=n;i++){
scanf("%d%d%d",&s[i].x,&s[i].y,&s[i].c);
b[++tot]=s[i].x;b[++tot]=s[i].y;
}
sort(b+1,b+tot+1);sort(s+1,s+n+1);
int top=unique(b+1,b+tot+1)-b-1;
for(int i=1;i<=n;i++){
s[i].x=lower_bound(b+1,b+top+1,s[i].x)-b;
s[i].y=lower_bound(b+1,b+top+1,s[i].y)-b;
}
int r1,r2,flag=0;
for(int i=1;i<=n;i++){
r1=zh(s[i].x);r2=zh(s[i].y);
if(s[i].c && r1!=r2) fa[r1]=r2;
else if(!s[i].c && r1==r2) {flag=1;break;}
}
if(flag) printf("NO\n");
else printf("YES\n");
}
return 0;
}