HDU 5253 最小生成树(kruskal)+ 并查集

本文介绍了一种基于最小生成树算法解决特定问题的实现方法。通过C++代码详细展示了如何构造边集合,并采用按权值排序的方式进行遍历,确保最终生成的树能够连接所有顶点且不存在环路,同时总权重最小。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目链接


#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define N 1000010
using namespace std;
struct node
{
    int a;//端点1
    int b;//端点2
    int num;//权值
};
bool cmp(node x,node y)//按权值升序
{
    return x.num<y.num;
}
int maps[1003][1003];
node edge[N*2];//边数=2mn-m-n<2mn
int father[N];
int r[N];
int finds(int x)
{
    if(father[x]!=x)
    father[x]=finds(father[x]);
    return father[x];
}
void connect(int a,int b)
{
    if(r[a]>r[b])
    father[b]=a;
    else if(r[a]<r[b])
    father[a]=b;
    else
    {
        father[a]=b;
        r[b]++;
    }
}
int main()
{
    int T,cas=1,n,m,ans,id;
    int i,j;
    scanf("%d",&T);
    while(T--)
    {
        int numedge=0;//边数
        ans=0;
        memset(r,0,sizeof(r));
        scanf("%d%d",&n,&m);
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=m;j++)
            scanf("%d",&maps[i][j]);
        }
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=m;j++)
            {
                id=(i-1)*m+j;
                father[id]=id;
                if(i<n)//除最后一行外
                {
                    numedge++;
                    edge[numedge].a=id;
                    edge[numedge].b=i*m+j;
                    edge[numedge].num=abs(maps[i][j]-maps[i+1][j]);
                }
                if(j<m)//除最后一列外
                {
                    numedge++;
                    edge[numedge].a=id;
                    edge[numedge].b=(i-1)*m+j+1;
                    edge[numedge].num=abs(maps[i][j]-maps[i][j+1]);
                }
            }
        }
        sort(edge+1,edge+numedge+1,cmp);//权值排序
        for(i=1;i<=numedge;i++)
        {
            int x=finds(edge[i].a);
            int y=finds(edge[i].b);
            if(x!=y)
            {
                connect(x,y);
                ans+=edge[i].num;
            }
        }
        printf("Case #%d:\n%d\n",cas++,ans);
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值