zoj - 1789 - The Suspects

题意：有n个学生，编号为0到n-1，m个群，给出每个群的学生，开始学生0被怀疑得了SARS，于是，人传人，被怀疑的学生所属群的所有学生都被怀疑，问共有几个学生被怀疑。

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=789

——>>简单的并查集题目，将每个群的后一个学生与前一个学生judge，统计结点数输出即可。

带路径压缩递归写法：

#include <iostream>

using namespace std;

const int maxn = 30000 + 10; //0 < n <= 30000，开30000+10够矣

int fa[maxn], height[maxn], cnt[maxn]; //fa[i]为i的父亲，height[i]为以i为根的树的高度，cnt[i]为以i为根的树的叶子的结点数

int find(int x) //找根
{
return fa[x] == x ? x : (fa[x] = find(fa[x]));
}

bool judge(int x, int y) //判断、合并操作
{
int new_x = find(x); //找x的根
int new_y = find(y); //找y的根
if(new_x == new_y) return 0;
fa[new_y] = new_x; //将树new_y合并到new_x中
cnt[new_x] += cnt[new_y]; //更新树new_x的叶子数
return 1;
}

int main()
{
int n, m, i, j, num, first, temp; //num为一个GROUPR的人数，first为一个GROUP的第一个人
while(cin>>n>>m)
{
if(n == 0 && m == 0) return 0;
for(i = 0; i < n; i++) //初始化
{
fa[i] = i; //建立n棵树，自己就是树根
cnt[i] = 1; //只有自己，结点数为1
}
for(i = 0; i < m; i++)
{
cin>>num>>first; //先输入第一个人，后面的人每一个都与前一个judge
for(j = 0; j < num-1; j++)
{
cin>>temp;
judge(first, temp);
}
}
cout<<cnt[find(0)]<<endl;
}
return 0;
}


不那么地压缩写法：

#include <iostream>

using namespace std;

const int maxn = 30000 + 10;        //0 < n <= 30000，开30000+10够矣

int fa[maxn], height[maxn], cnt[maxn];      //fa[i]为i的父亲，height[i]为以i为根的树的高度，cnt[i]为以i为根的树的叶子的结点数

int find(int x)     //找根
{
while(x != fa[x])
{
x = fa[x];
}
return x;
}

bool judge(int x, int y)        //判断、合并操作
{
int new_x = find(x);        //找x的根
int new_y = find(y);        //找y的根
if(new_x == new_y) return 0;
else
{
if(height[new_x] > height[new_y])       //当树new_x的高度 > 树new_y的高度时
{
fa[new_y] = new_x;      //将树new_y合并到new_x中
cnt[new_x] += cnt[new_y];       //更新树new_x的叶子数
}
else if(height[new_x] == height[new_y])     //当树new_x的高度 == 树new_y的高度时
{
fa[new_y] = new_x;      //将树new_y合并到new_x中
cnt[new_x] += cnt[new_y];       //更新树new_x的叶子数
height[new_x]++;        //树new_x的高度加1
}
else        //当树new_x的高度 == 树new_y的高度时
{
fa[new_x] = new_y;      //将树new_x合并到new_y中
cnt[new_y] += cnt[new_x];       //更新树new_y的叶子数
}
return 1;
}
}

int main()
{
int n, m, i, j, num, first, temp;       //num为一个GROUPR的人数，first为一个GROUP的第一个人
while(cin>>n>>m)
{
if(n == 0 && m == 0) return 0;
for(i = 0; i < n; i++)      //初始化
{
fa[i] = i;      //建立n棵树，自己就是树根
cnt[i] = 1;     //只有自己，结点数为1
height[i] = 1;      //只有自己，为1层
}
for(i = 0; i < m; i++)
{
cin>>num>>first;        //先输入第一个人，后面的人每一个都与前一个judge
for(j = 0; j < num-1; j++)
{
cin>>temp;
judge(first, temp);
first = temp;       //更新前者
}
}
cout<<cnt[find(0)]<<endl;
}
return 0;
}