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;  
}