POJ1611 The Suspects

木白星枝

于 2019-04-01 12:23:33 发布

阅读量162

点赞数

CC 4.0 BY-SA版权

分类专栏：并查集

本文链接：https://blog.youkuaiyun.com/qq_42759120/article/details/88945324

并查集专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一个经典的并查集应用案例，通过并查集数据结构解决学生间疾病传播的问题。具体而言，针对n个学生分属m个团体的情况，当其中一个学生疑似患病时，如何高效地判断出所有疑似患病的学生数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

这是一个模板题：

#include<iostream>
#include<algorithm>
#include<fstream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<string>
#include<cmath>
#include<cctype>
#include<vector>
#include<limits.h>
#include<queue>
    
using namespace std;

int par[30004];
int total[30004];

int GetRoot(int a)  //找该节点的父节点，并且合并
{
    if(par[a]!=a)
    {
        par[a]=GetRoot(par[a]);
    }
    return par[a];
}

void Merge(int a,int b) //合并2个节点或集合
{
    int c1,c2;
    c1=GetRoot(a);
    c2=GetRoot(b);
    if(c1==c2)
        return ;
    par[c2]=c1;
    total[c1]+=total[c2];
}

int main()
{
    int n,m,i,j,a,b,k,e;
    while(scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0)
            break;
        for(i=0; i<n; ++i)
        {
            par[i]=i;
            total[i]=1;
        }
        for(i=0; i<m; ++i)
        {
            scanf("%d%d",&k,&a);
            for(j=1; j<k; ++j)
            {
                scanf("%d",&b);
                Merge(a,b);
            }
        }
        e=GetRoot(0);
        printf("%d\n",total[e]);    //这里不能直接用e，是因为可能还没进行过路径压缩
    }
    return 0;
}