【简单并查集】算法学习

---

并查集思想：

并查集是一种树型的数据结构，用于处理一些不相交集合（Disjoint Sets）的合并及查询问题。我们通常是在开始时让每个元素构成一个单元素的集合，然后按一定顺序将属于同一组的元素所在的集合合并，其间要反复查找一个元素在哪个集合中。通过并查集，可以有效降低这一类问题的时间和空间复杂度。

---

并查集实现：

一、并查集首先要定义一个数组f（也可以用结构体实现），用来储存i所在的集合。

• 如果f[i]=i，则说明i是根节点，属于自己的集合。
• 如果f[i]=j，则说明i的父节点为j。

二、并查集初始化：

void UFInit()
{
    for(int i=0;i<n;i++)
        f[i]=i;
}

三、事实上，并查集主要包含两个操作：合并和查询。所以这也是称之为并查集的原因。

查询操作（版本一）：

int Find(int a)
{
    if(f[a]==a)
        return a;

    else
    {
        f[a]=Find(f[a]);    //该步为路径压缩，可降低时间复杂度。
        return f[a];
    }
}

查询操作（版本二）（推荐使用，这样可以提高书写代码效率）：

int Find(int a)
{
    return f[a]==a? a:f[a]=Find(f[a]);
}

合并操作：

void Union(int a,int b)
{
    a=Find(a);
    b=Find(b);

    if(a!=b)
        f[a]=f[b];
}

---

例题：

题目地址：The Suspects POJ-1611

题目大意：n个学生分属m个团体。一个学生如果患病，则他所属的整个团体都疑似患病。已知0号学生疑似患病，以及每个团体都由哪些学生构成，求疑似患病的学生总数。

代码如下：

#include <iostream>
#include <cstdio>

using namespace std;

const int N=30001;
int f[N+1];
int n,m;

void UFInit()
{
    for(int i=0;i<n;i++)
        f[i]=i;
}

int Find(int a)
{
    if(f[a]==a)
        return a;

    else
    {
        f[a]=Find(f[a]);    //路径压缩很有必要，否则容易超时
        return f[a];
    }
}

void Union(int a,int b)
{
    a=Find(a);
    b=Find(b);

    if(a!=b)
        f[a]=f[b];
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF&&n)
    {
        int a,t,k,num=0;
        UFInit();

        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&k,&a);

            for(int j=1;j<k;j++)
            {
                scanf("%d",&t);
                Union(a,t);
            }
        }

        for(int i=0;i<n;i++)
            if(Find(f[0])==Find(f[i]))
                num++;

        printf("%d\n",num);
    }

    return 0;
}