[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[],a[],b[],op[];

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

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

         for(i=;i<=n;++i)scanf("%d%d%d",&a[i],&b[i],&op[i]);

         for(i=;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=;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\n");
                     goto Fail;
                 }
             }
         }

         printf("YES\n");
 Fail:;
     }

     return ;
 }