输入样例:
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1
输出样例:
NO
YES
解析:
并查集,由于 i, j 数据量为1e9,所以需要离散化。
数组要开2e5个,由于可能每条“约束条件”的数据都不一样。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
int t,n,s[N],idx,flag;
map<int,int>mp;
int func(int x){
if(mp.count(x)) return mp[x];
return mp[x]=++idx;
}
int find(int x){
if(x!=s[x]) s[x]=find(s[x]);
return s[x];
}
void merge(int x,int y){
x=find(x),y=find(y);
if(x!=y) s[x]=y;
}
struct node{
int x,y,e;
}a[N];
int main(){
scanf("%d",&t);
while(t--){
flag=1,idx=0;
mp.clear();
scanf("%d",&n);
for(int i=1;i<=200000;i++) s[i]=i;
for(int i=0;i<n;i++){
int x,y,e;
scanf("%d%d%d",&x,&y,&e);
x=func(x);
y=func(y);
if(e) merge(x,y);
a[i]={x,y,e};
}
for(int i=0;i<n;i++){
if(!a[i].e)
if(find(a[i].x)==find(a[i].y)){
flag=0;
cout<<"NO"<<endl;
break;
}
}
if(flag) cout<<"YES"<<endl;
}
return 0;
}