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