poj1611 The Suspects 并查集

poj1611 The Suspects 并查集

标签：并查集

题目链接

/*
    题意：n个学生(0~n-1)，m组，一个学生可以同时加入不同的组。
          有一种传染病，如果一个学生被感染，那么和他同组的学生都会被感染。
          现已知0号学生被感染，问一共有多少个人被感染。

    思路：基本的并查集操作。
          首先将每个学生都初始化为一个集合，然后将同组的学生合并，
          设置一个数组num[]来记录每个集合中元素的个数，最后只要输出0号学生所在集合中元素的个数即可。
*/
#include <stdio.h>

int num[30005];  //结点所在集合元素的个数
int father[30005];

void make_set(int x){  //初始化集合
    father[x] = x;
    num[x] = 1;
}

int find_set(int x){  //查找x元素所在集合, 返回根节点
    return  x == father[x] ? father[x] : find_set(father[x]);
}

void union_(int x, int y){  //合并x, y所在的集合
    x = find_set(x), y = find_set(y);
    if(x == y)  return ;  //同一个集合则不需要合并
    if(num[x] < num[y]) {  //小集合合并到大集合
        father[x] = y;
        num[y] += num[x];
    }
    else{
        father[y] = x;
        num[x] += num[y];
    }
}

int main(){
    int n, m, t, x, y, i, j;

    while(scanf("%d %d", &n, &m) && (n || m)){
        for(i = 0; i < n; i++)  make_set(i);
        for(i = 0; i < m; i++){
            scanf("%d", &t);
            scanf("%d", &x);
            for(j = 1; j < t; j++){
                scanf("%d", &y);
                union_(x, y);
            }
        }
        printf("%d\n", num[find_set(0)]);
    }

    return 0;
}