POJ 1611 - The Suspects

Advanced Data Structures :: Disjoint Set

Description

有n个学生，编号从 0 到 n - 1 。

学生们分成了几个团伙。

而学生0是个“咸鱼饭”，因此和他一个团伙的人都是“咸鱼饭”。

问有几个“咸鱼饭”。

Type

Advanced Data Structures :: Disjoint Set

Analysis

用并查集并一下，然后枚举下每个学生，

查一下有多少人和学生0一个集合，就可以了。

Solution

// POJ 1611
// The Suspect
// by A Code Rabbit

#include <cstdio>

const int MAXN = 30002;

struct DisjointSet {
    int p[MAXN];
    void Init(int);
    void Make(int x) { p[x] = x; }
    int Find(int x) { return p[x] == x ? x : p[x] = Find(p[x]); }
    int Union(int x, int y) { p[Find(x)] = Find(y); }
};

void DisjointSet::Init(int n) {
    for (int i = 0; i < n; i++)
        Make(i);
}

int n, m;
DisjointSet set;

int main() {
    while (scanf("%d%d", &n, &m) && (n || m)) {
        set.Init(n);
        for (int i = 0; i < m; i++) {
            int k, x, y;
            scanf("%d", &k);
            scanf("%d", &x);
            for (int j = 0; j < k - 1; j++) {
                scanf("%d", &y);
                set.Union(x, y);
            }
        }
        int ans = 0;
        for (int i = 0; i < n; i++)
            if (set.Find(i) == set.Find(0))
                ans++;
        printf("%d\n", ans);
    }

    return 0;
}