poj1611-The Suspects(并查集)


题目描述:

有很多组学生,在同一个组的学生经常会接触,也会有新的同学的加入。但是SARS是很容易传染的,只要在改组有一位同学感染SARS,那么该组的所有同学都被认为得了SARS。现在的任务是计算出有多少位学生感染SARS了。假定编号为0的同学是得了SARS的。


并查集,不压缩路径会wa,没啥好说的。


AC代码:


#include <iostream>
#include <cstring>
using namespace std;
int n,m;
int pre[30010];
int nums[30010];
int find(int x)  //不压缩路径会wa
{
    if(pre[x]==x)
        return x;
    else
        return pre[x]=find(pre[x]);
}
void join(int x,int y) //并查集的join函数
{
    int rootx=find(x);
    int rooty=find(y);
    if(rootx!=rooty)
        pre[rootx]=rooty;
}
int main()
{
    while(cin>>n>>m)
    {
        if(n==0&&m==0) break;
        for(int i=0;i<n;i++)
            pre[i]=i;
        for(int i=0;i<m;i++)
        {
            int num;
            cin>>num;
            if(num>=1){
            for(int j=0;j<num;j++)
            {
                cin>>nums[j];
            }
            for(int j=1;j<num;j++)
            {
                join(nums[0],nums[j]); //把num[0]当成这个小组的老大
            }
            }
        }
        int ans=0;
        for(int i=0;i<n;i++) //如果一个人的老大和0的老大是一样的,说明这个人生病
        {
            if(find(i)==pre[0])
                ans++;
        }
        cout<<ans<<endl;
    }
    return 0;
}

阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页