HDU 1232 畅通工程 (并查集)

一题并查集的题目,采用并查集的方式。

AC代码如下:

#include <iostream>
#include <set>
#include <cstring>
using namespace std;
const int maxn=1e3+5;
int pre[maxn]; //存储该节点的父节点
int son[maxn]; //用来存储该节点中结点个数
int Find(int x) //查找父节点
{
    int r=x;
    while(pre[r]!=r)
        r=pre[r];
    pre[x]=r;
    return r;
}
void Union(int x,int y)
{
    int fx,fy;
    fx=Find(x);
    fy=Find(y);
    if(fx!=fy)  //这里每次都合并子节点个数多的以便减少修路的次数
    {
        if(son[fx]<son[fy])
        {
            pre[fx]=fy;
            son[fy]+=son[fx];
            son[fx]=0;
        }
        else
        {
            pre[fy]=fx;
            son[fx]+=son[fy];
            son[fy]=0;
        }
    }

    return ;
}
int main()
{
    int N,M;
    while(cin>>N && N)
    {
        memset(son,1,sizeof(son)); //先使子节点个数都处理为1
        cin>>M;
        for(int i=1;i<=N;i++)   //初始化父节点
            pre[i]=i;
        int x,y;
        for(int i=0;i<M;i++)
        {
            cin>>x>>y;
            Union(x,y); //两点之间修路
        }
        int sum=0;
        for(int i=1;i<=N-1;i++)
        {
            if(Find(i)!=Find(i+1)) //问还要修多少条路
            {
                sum++;
                Union(i,i+1);
            }
        }

        cout<<sum<<endl; //最后输出路的条数
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43508782/article/details/84933390