POJ-1611 The Suspects (并查集)

POJ-1611 The Suspects (并查集)

原题链接：http://poj.org/problem?id=1611

题目大意：

将学生逐一编号，并分成 k 个小组，有传染病SARS,只要小组中有传染源，则整个小组的人均被感染，求最终感染者人数。（注：起始 0 为感染源）

分析：

根据题目的大意易知，只要与 0 号学生有关系的都为感染者，即与 0 号学生同一组的人数即感染者人数。易想到 并查集 和 建树遍历 。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int father[30010];

int root(int x) //找到 x 的根节点
{
while (father[x]!=x) //while -- 找到根节点的值， 若改为 if 只能找到上一个节点的值
x=father[x];
return x;
}
void build(int x,int y)
{
int x1,y1;
x1=root(x);
y1=root(y);
if (x1!=y1)
{
if (x1==0) father[y1]=x1; //若 x 的根节点 x1 为 0 ，则 y 的根节点与 x 建立关系，其根节点也应为 0
else father[x1]=y1;
}
}
int main()
{
int n,m;
while (cin>>n>>m)
{
memset (father,0,sizeof(father));

if (n==0&&m==0) return 0;

for(int i=0;i<n;i++) father[i]=i;//学生逐一编号

while (m--)
{
int k,a;
cin>>k>>a;// a 为每一行的根节点
for (int i=1;i<k;i++)
{
int b;cin>>b;
build(a,b);
4000
//逐一对行内的点建立连接
}
}
int ans=0;
for (int i=0;i<n;i++)
{
if (root(father[i])==0) ans++;
}
cout<<ans<<endl;
}
}