POJ-1611 The Suspects

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

题目大意：

有一个学校，有N个学生，编号为0-N-1，现在0号学生感染了非典，凡是和0在一个社团的人就会感染，并且这些人如果还参加了别的社团，他所在的社团照样全部感染，求感染的人数。

解题思路：

并查集的变种，实质就是求0所在的强连通图的结点数目。

这道题纠结在数据的输入上，他只是告诉你哪些学生是同一个社团的。这就需要处理一下，我的想法是：如果这个社团有num个孩子，new出一个大小为num的数组，第一个孩子不处理，从第二个孩子起，和上个孩子合并，这样就完成了他们的合并。

代码如下：

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 30010
int pre[N], son[N];

int find(int x)
{
	return x == pre[x] ? x : find(pre[x]);
}

void join(int x, int y)
{
	int root1, root2;
	root1 = find(x);
	root2 = find(y);
	if(root1 != root2)
	{
			pre[root2] = root1;
			son[root1] += son[root2];
	}
}

int main()
{
	int n, m, num;
	while(scanf("%d%d", &n, &m) && n + m)
	{
		for(int i = 0; i < n; ++i)
		{
			pre[i] = i;
			son[i] = 1;
		}
		for(int i = 1; i <= m; ++i)
		{
			scanf("%d", &num);
			int *stu = new int[num]; //YY出来的~~~
			for(int j = 0; j < num; ++j)
			{
				scanf("%d", &stu[j]);
				if(j != 0)
					join(stu[j - 1], stu[j]);
			}
			delete(stu);
		}
		printf("%d\n", son[find(0)]);
	}
	return 0;
}