HDU1232畅通工程（并查集）

链接：HDU1232畅通工程
题意：某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要互相间接通过道路可达即可）。问最少还需要建设多少条道路
思路：并查集，把所有能联通的路搞成一组 ，然后搜索一共有x组，那就还需要x-1条路

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int n,m,a,b,sum;
int fa[10000];
int main()
{
    while(cin>>n&&n!=0)
    {
        sum=0;
        memset(fa,0,sizeof(fa));
        cin>>m;
        for(int i=1;i<=n;i++)
        {
            fa[i]=i;
        }//初始化数组使每个地方的父节点都先是他自己
        for(int i=0;i<m;i++)//输入m组路的关系
        {
            cin>>a>>b;
            while(a!=fa[a])
                a=fa[a];//找到a的最...emmm上面的点
            while(b!=fa[b])
                b=fa[b];//b的
            if(a!=b)
                fa[a]=b;//使a的最上点连在b的最上点上...就是把跟a通的和跟b通的连为一组
        }
        for(int i=1;i<=n;i++)
        {
            if(fa[i]==i)
            {
                ++sum;
                //cout<<i<<endl;
            }//一组中只能有一个数自己是自己的父节点...就是最上面那个数，所以数一下一共几个

        }
        cout<<sum-1<<endl;
    }
}

完了以后，发现用函数写可能更好所以改进了一下代码

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int n,m,a,b,sum;
int fa[10000];
int ifind(int x)
{
  while(x!=fa[x])
        x=fa[x];
  return x;
}
void combine(int x,int y)
{
    int a,b;
    a=ifind(x);
    b=ifind(y);
    if(a!=b)
    fa[a]=b;
}
int main()
{
    while(cin>>n&&n!=0)
    {
        sum=0;
        memset(fa,0,sizeof(fa));
        cin>>m;
        for(int i=1;i<=n;i++)
        {
            fa[i]=i;
        }
        for(int i=0;i<m;i++)
        {
            cin>>a>>b;
            combine(a,b);
        }
        for(int i=1;i<=n;i++)
        {
            if(fa[i]==i)
            {
                ++sum;
            }

        }
        cout<<sum-1<<endl;
    }
    return 0;
}

可以在combine函数里加一段压缩路径的代码

while(i!=r)
{
    int temp=fa[i];
    fa[i]=r;
    i=temp;
}