poj1611 The Suspects

题意：

n个学生分属m个团体，(0 < n <= 30000 ,0 <= m <= 500) 一个学生可以属于多个团体。一个学生疑似患病，则它所属的整个团体都疑似患病。已知0号学生疑似患病，以及每个团体都由哪些学生构成，求一共多少个学生疑似患病。

Sample Input

100 4
2 1 2
5 10 13 11 12 14
2 0 1
2 99 2
200 2
1 5
5 1 2 3 4 5
1 0
0 0

Sample Output

4
1
1

思路：

并查集。

实现：

#include <iostream>
#include <cstdio>
using namespace std;

int par[30005];
int ran[30005];

void init(int n)
{
    for (int i = 0; i < n; i++)
    {
        ran[i] = 0;
        par[i] = i;
    }
}

int find(int x)
{
    if (par[x] == x)
        return x;
    return par[x] = find(par[x]);
}


void unite(int x, int y)
{
    x = find(x);
    y = find(y);
    if (x == y)
        return;
    if (ran[x] < ran[y])
        par[x] = y;
    else
    {
        par[y] = x;
        if (ran[x] == ran[y])
            ran[x]++;
    }
}

int n, m, p, x, y;
int main()
{
    while (cin >> n >> m, n || m)
    {
        init(n + 1);
        for (int i = 0; i < m; i++)
        {
            scanf("%d", &p);
            if (!p)
                continue;
            scanf("%d", &x);
            for (int j = 0; j < p - 1; j++)
            {
                scanf("%d", &y);
                unite(x, y);
            }
        }
        int tmp = find(0);
        int cnt = 0;
        for (int j = 0; j < n; j++)
        {
            if (find(j) == tmp) //不能写成par[j],因为点j可能还没进行路径压缩，结果不一定正确。
                cnt++;
        }
        printf("%d
", cnt);
    }
    return 0;
}