题目大意:有编号为0~n-1的n个人,其中编号0中了SARS,如果跟他同一组就也会得病,问一共有多少人得病。输入m组,每组的成员。只有有同一个组员,则两个组是同一个组。
解题思路:并查集基础,只是因为0号得病,所以0号作为最上级的话便于查找,所以在合并时要考虑是否其中某个的上级为0,如果有,则合并时注意将另一个的上级的上级改为0。
ac代码:
#include <iostream>
using namespace std;
int n, m, a[30005], temp, pre[30005], sum;
int find(int x)
{
int r=x, temp;
while (r != pre[r])
r = pre[r];
while (r != x){
temp = pre[x];
pre[x] = r;
x = temp;
}
return r;
}
void join(int x, int y)
{
int a=find(x), b=find(y);
if (!b && a != b)
pre[a] = b;
else if (a != b)
pre[b] = a;
}
int main()
{
while (scanf("%d%d", &n, &m)!=EOF && n+m){
sum = 1;
for (int i=0; i<n; i++)
pre[i] = i;
for (int i=0; i<m; i++){
scanf("%d", &temp);
for (int j=0; j<temp; j++){
scanf("%d", &a[j]);
if (j)
join(a[j-1], a[j]);
}
}
for (int i=1; i<n; i++)
if (!find(i))
sum++;
printf("%d\n", sum);
}
return 0;
}