并查集

并查集

先看看例题：

**
有n个人，编号1-n。
现在有一个舞会，在舞会上，大家会相互介绍自己的朋友。
即： 如果a认识b，b认识c。那么在舞会上，a就会通过b认识到c。
现在，给出m个关系
每个关系描述：
a b
表示 编号为a和编号为b的人是朋友关系。

格式
输入格式
输入n和m
接下来m行，每行为a b

输出格式
最后问，会有多少个朋友圈。

样例输入 Copy
5 3
1 2
2 3
4 5
样例输出 Copy
2
**

像这种题，就需要用到并查集。首先将数组中的每一个元素都等于其下标：

for(i=1; i<=n; i++)
            a[i]=i;

这样，每一个人都有其最初的关系。

然后，就样例而言，关系中不乏诸如1–2 2–3 3–4等等长串。这会使输出的朋友圈个数不定（具体为何博主也不是很懂，总之答案会错误50%）。因此，我们需要路径压缩。

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

即直接将关系变为1–2 1–3 1–4

因此，得代码如下：

#include<stdio.h>
int a[1000010]= {0};
int f(int x)
{
    if(x==a[x]) return x;
    a[x]=f(a[x]);
    return a[x];
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        int cnt=0,i;
        for(i=1; i<=n; i++)
            a[i]=i;//将每个人之间的关系初始化
        for(i=0; i<m; i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            a[f(x)]=f(y);//路径压缩
        }
        for(i=1; i<=n; i++)
            if(a[i]==i)//找根节点，根节点的个数即为朋友圈个数
                cnt++;
        printf("%d\n",cnt);
    }
    return 0;
}