[Bzoj4195] [NOI2015] 程序自动分析 [并查集，哈希，map] 题解

用并查集+离散化，注意：并查集数组大小不是n而是n*2

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>
#include <map>

using namespace std;

int    fa[2100000],a[1100000],b[1100000],op[1100000];

inline    int    get_anc(const int x)
{
    return fa[x]==x ? x:fa[x]=get_anc(fa[x]);
}

int main()
{
    int    i,n,q,S,T,cnt=0;

    scanf("%d",&q);
    while(q--)
    {
        scanf("%d",&n);
        map<int,int>    Map;
        Map.clear();cnt=0;
        for(i=1;i<=n<<1;++i)fa[i]=i;
        
        for(i=1;i<=n;++i)scanf("%d%d%d",&a[i],&b[i],&op[i]);

        for(i=1;i<=n;++i)
        {
            if(op[i])
            {
                if(!Map[a[i]])Map[a[i]]=++cnt;
                if(!Map[b[i]])Map[b[i]]=++cnt;
                int x=Map[a[i]],y=Map[b[i]];
                S=get_anc(x),T=get_anc(y);
                if(S!=T)fa[S]=T;
            }
        }

        for(i=1;i<=n;++i)
        {
            if(!op[i])
            {
                if(!Map[a[i]])Map[a[i]]=++cnt;
                if(!Map[b[i]])Map[b[i]]=++cnt;
                int x=Map[a[i]],y=Map[b[i]];
                S=get_anc(x),T=get_anc(y);
                if(S==T)
                {
                    printf("NO
");
                    goto Fail;
                }
            }
        }
        
        printf("YES
");
Fail:;
    }

    return 0;
}