num.11The Suspects POJ - 1611

题解:

题目的大概意思是,n个人,如果有一个人感染了SARS病毒,那么和这个人同组的人都会感染病毒。也就是说和被感染的人有关系的人都感染了病毒。
输入:n m n为成员个数 m为组数
输出:被感染的人数

代码

#include
#include
using namespace std;
int pre[30050];//定义父亲节点数组
int Find(int x){
     //注意:这里必须将所有询问到第一个节点的都变成它的子节点,即将父节点统一
    int r=x;
    while(pre[r]!=r)
        r=pre[r];
    int i=x,j;
    while(pre[i]!=r){
     
        j= pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}

void join(int x, int y){
     //合并两个集合
    int fx = Find(x);
    int fy = Find(y);
    if(fx!= fy)
        pre[fx] = fy;
    return ;
}

int main(){
     
    int n, m, a, i, j, num;
    int arr[30020];
    while(scanf("%d%d", &n, &m), n+m){
     
        num = 0;
        for(i=0;i<n;i++){
     
            pre[i]=i;
        }
        while(m--){
     
            scanf("%d",&a);
            scanf("%d",&arr[0]);
            for(i=1;i<a;i++){
     
                scanf("%d",&arr[i]);
                join(arr[0],arr[i]);
            }
        }
        for(i=0;i<n;i++){
     
            if(Find(i)==pre[0])
                num++;
        }
        printf("%d\n",num);
    }
    return 0;
}

Ps:

并查集的最基本也是最重要,最核心的三个操作:找该元素所在集合的根节点、合并两个集合以及判断两个元素是否属于一个集合。

你可能感兴趣的:(并查集)