题目描述:
有很多组学生,在同一个组的学生经常会接触,也会有新的同学的加入。但是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;
}