我是最近刚刚开始学习并查集的。有些事情可能还不是特别明白。

现在求助的这道题，我就是用并查集做的。

思路

1. 首先按照要求读入，若遇到$x_i=x_j$并且$e=0$的情况直接输出NO
2. 当数据活过上述情况时，对所有数据按照$e$从大到小排序，即$e=1$的数据在前面
3. 若某一行数据形似$x_i\ x_j\ 1$，那么将$x_i$设为$x_j$的父亲
4. 若某一行数据形似$x_i\ x_j\ 0$，那么进行查找操作，若$x_i$和$x_j$没有公共祖先，那么输出NO
5. 当数据活过上述所有情况时，输出YES

【代码】

#include<bits/stdc++.h>
using namespace std;

struct num{
	int x,y,e;
};
int fa[1000010];

bool cmp(num p,num q)
{
	return p.e > q.e;
}

int find(int x)
{
	if(fa[x]==x) return x;
	fa[x]=find(fa[x]);
	return fa[x];
}

int main()
{
    int t;
	 cin>>t;
	 while(t--)
	 {
	    bool flag=true,flag2=true;
    	int n;
    	num a[1000005];
    	cin>>n;
    	for(int i=1;i<=n;i++)
    	{
    		cin>>a[i].x>>a[i].y>>a[i].e;
    		if(a[i].x == a[i].y && a[i].e==0)
    		{
    			cout<<"NO";
    			flag=false;
    			break;
    		}
    	}
    	sort(a+1,a+n+1,cmp);
    	for(int i=1;i<=n && flag==true;i++)
    	{
    		int xx=find(a[i].x);
    		int yy=find(a[i].y);
    		if(a[i].e==1)
    			if(xx!=yy)
    				fa[a[i].y]=a[i].x;
    		else if(xx==yy)
    		{
    			cout<<"NO";
    			flag2=false;
    			break;
    		}
    	}
    	if(flag==true && flag2==true) cout<<"YES";
    	cout<<endl;
	}
	return 0;
}