POJ-2492 A Bug's Life（种类并查集）

http://poj.org/problem?id=2492

题意：

给出一个T代表几组数据,给出一个n一个m,代表人的编号由1~n,m条命令,每条命令由两个数值组成,代表这两个人性别不同,问所有命令是否符合逻辑

两种写法：

第一种：带权并查集

 #include <stdio.h>
 #include <string.h>
 #include <iostream>
 #include <string>
 #include <math.h>
 #include <algorithm>
 #include <vector>
 #include <stack>
 #include <queue>
 #include <set>
 #include <map>
 #include <sstream>
 const int INF=0x3f3f3f3f;
 typedef long long LL;
 const int mod=1e9+;
 const int maxn=1e5+;
 using namespace std;

 int fa[];
 int dis[];//表示i和其根结点的关系，1表示异性，0表示同性 

 void init(int n)
 {
     for(int i=;i<=n;i++)
         dis[i]=,fa[i]=i;
 }
 int Find(int x)//带权并查集
 {
     if(x!=fa[x])
     {
         int t=fa[x];
         fa[x]=Find(fa[x]);
         dis[x]=(dis[x]+dis[t])%;
     }
     return fa[x];
 }

 int main()
 {
     #ifdef DEBUG
     freopen("sample.txt","r",stdin);
     #endif
 //    ios_base::sync_with_stdio(false);
 //    cin.tie(NULL);

     int T;
     scanf("%d",&T);
     for(int k=;k<=T;k++)
     {
         int n,m;
         scanf("%d %d",&n,&m);
         init(n);
         int flag=;//flag=1表示找到了错误
         for(int i=;i<=m;i++)
         {
             int a,b;
             scanf("%d %d",&a,&b);
             if(flag) continue;
             int aa=Find(a);
             int bb=Find(b);
             if(aa==bb)// 二者的祖先相同
             {
                 if(dis[a]==dis[b]) flag=;//且与祖先的性别关系相同,出现错误
             }
             else//二者祖先不同
             {
                 fa[bb]=fa[aa];
                 dis[bb]=(dis[a]+-dis[b])%;
             }
         }
         if(k!=) printf("\n");//格式要求
         printf("Scenario #%d:\n",k);
         if(flag==) printf("Suspicious bugs found!\n");
         else printf("No suspicious bugs found!\n");
     }

     return ;
 }

第二种：

fa开两倍空间，fa[i+n]相当于i的对立集合，这种可能耗时会多一点吧

 #include <stdio.h>
 #include <string.h>
 #include <iostream>
 #include <string>
 #include <math.h>
 #include <algorithm>
 #include <vector>
 #include <stack>
 #include <queue>
 #include <set>
 #include <map>
 #include <sstream>
 const int INF=0x3f3f3f3f;
 typedef long long LL;
 const int mod=1e9+;
 const int maxn=1e5+;
 using namespace std;

 int fa[];
 void init(int n)
 {
     for(int i=;i<=n;i++)
         fa[i]=i;
 }
 int Find(int x)
 {
     return x==fa[x]? x:fa[x]=Find(fa[x]);
 }
 void Union(int a,int b)
 {
     int aa=Find(a);
     int bb=Find(b);
     if(aa!=bb) fa[aa]=bb;
 }

 int main()
 {
     #ifdef DEBUG
     freopen("sample.txt","r",stdin);
     #endif
 //    ios_base::sync_with_stdio(false);
 //    cin.tie(NULL);

     int T;
     scanf("%d",&T);
     for(int k=;k<=T;k++)
     {
         int n,m;
         scanf("%d %d",&n,&m);
         init(*n);
         int flag=;//flag=1表示找到了错误
         for(int i=;i<=m;i++)
         {
             int a,b;
             scanf("%d %d",&a,&b);
             if(flag) continue;
             int aa=Find(a);
             int bb=Find(b);
             if(aa==bb) flag=;// 二者同属一个集合，性别相同
             else//二者性别不同，不属于同一个集合
             {
                 Union(a,b+n);//a和b的对立集合同属一个集合
                 Union(a+n,b);//b和a的对立集合同属一个集合
             }
         }
         if(k!=) printf("\n");//格式要求
         printf("Scenario #%d:\n",k);
         if(flag==) printf("Suspicious bugs found!\n");
         else printf("No suspicious bugs found!\n");
     }

     return ;
 }

-