POJ - 2492 - A Bug's Life（带权并查集）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/sugarbliss/article/details/89481576

题目链接：http://poj.org/problem?id=2492

题意：T组测试数据，每组数据有一个N，M表示N个人和M个关系，下面M行为关系，每组关系有两个数字a，b表示a喜欢b，问有没有同性恋？

思路：带权并查集，定义1为同性，0为异性，带权并查集主要是查找和合并函数中权值的关系，还有初始化权值数组为1，因为开始，都是同性。

#include <stdio.h>
#include <string.h>
using namespace std;
const int N = 1e6 + 7;
int f[N], w[N], n, m, flag;
void init()
{
    for(int i = 0; i <= n; i++)
        f[i] = i, w[i] = 1;
}
int findd(int x)
{
    if(f[x] == x) return x;
    int t = f[x];
    f[x] = findd(f[x]);
    w[x] = (w[x] + w[t] + 1) % 2;
    return f[x];
}
void unint(int x, int y)
{
    int tx = findd(x);
    int ty = findd(y);
    if(tx != ty)
    {
        f[tx] = ty;
        w[tx] = (-w[x] + w[y] + 2) % 2;
    }
    else
    {
        if((w[y] - w[x] + 2) % 2 == 0) flag = 1;
    }
}
int main()
{
    int t, cas = 1; scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n, &m);
        init();
        flag = 0;
        while(m--)
        {
            int a, b;
            scanf("%d%d",&a, &b);
            unint(a, b);
        }
        printf("Scenario #%d:\n",cas++);
        if(flag) puts("Suspicious bugs found!");
        else puts("No suspicious bugs found!");
        if(t) printf("\n");
    }
}