写的是非离散化的70pts,实际50pts。
#include<bits/stdc++.h>
using namespace std;
int t,n,fa[1000005],size[1000005];
struct node{
int x,y;
bool e;
}a[1000005];
int find(int x){
return x==fa[x]?x:fa[x]=find(fa[x]);
}
void join(int x,int y)
{
int f1=find(x),f2=find(y);
if(f1!=f2){
if(size[f1]<size[f2])
swap(f1,f2);
fa[f2]=f1;
size[f1]+=size[f2];
}
}
bool cmp(node s,node t){
if(s.e==t.e){
if(s.x==t.x) return s.y<t.y;
return s.x<t.x;
}
return s.e>t.e;
}
void solve(){
memset(fa,0,sizeof fa);
memset(a,0,sizeof a);
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i].x>>a[i].y>>a[i].e,fa[i]=i;
sort(a+1,a+n+1,cmp);
for(int i=1;i<=n;i++){
int s=find(a[i].x),t=find(a[i].y);
if(a[i].e){
if(s!=t) join(s,t);
}else if(!a[i].e&&s==t){puts("NO");return;}
}
puts("YES");
return;
}
signed main(){
cin>>t;
for(;t;t--) solve();
return 0;
}