bzoj 4195: [Noi2015]程序自动分析【并查集】

等于有传递性，所以hash一下把等于用并查集连起来，然后再判断不等于是否合法即可

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<map>
using namespace std;
const int N=200005;
int T,n,x,y,v,f[N],g[N],tot,has;
bool fl;
map<int,int>mp;
struct qwe
{
    int a,b,o;
}a[N];
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
int zhao(int x)
{
	return x==f[x]?x:f[x]=zhao(f[x]);
}
int main()
{
    T=read();
    while(T--)
    {
        n=read();
        fl=0,tot=0,has=0;
        for(int i=1;i<=n;i++)
        {
            a[i].a=read(),a[i].b=read(),a[i].o=read();
            g[++tot]=a[i].a,g[++tot]=a[i].b;
        }
        sort(g+1,g+1+tot);
		for(int i=1;i<=tot;i++)
			if(i==1||g[i]!=g[i-1])
				mp[g[i]]=++has;
		for(int i=1;i<=n;i++)
			a[i].a=mp[a[i].a],a[i].b=mp[a[i].b];
        for(int i=1;i<=has;i++)
			f[i]=i;
        for(int i=1;i<=n;i++)
			if(a[i].o)
			{
				int fu=zhao(a[i].a),fv=zhao(a[i].b);
				if(fu!=fv)
					f[fu]=fv;
			}
        for(int i=1;i<=n&&!fl;i++)
			if(!a[i].o&&zhao(a[i].a)==zhao(a[i].b))
				fl=1;
        if(fl)
			puts("NO");
        else
			puts("YES");
    }
    return 0;
}