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