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; }