最小生成树实现(kruskal+并查集)续

题目:(和上篇的那个题目差不多,不过里用到了路径压缩,所以效率更高。还有就是数据后面输入的,所以,更上篇的那个题目还是有区别。)

输入数据:

具体实现代码(C++):

#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;

#define MAX 100

//定义边(x,y),权值为W
typedef struct
{
    int x,y;
    int w;
}edge;

edge e[MAX];
//rank[x]表示x所在树的高度
int rank[MAX];
// father[x]表示x的父节点
int father[MAX];
int sum;

//比较函数,按权值(相同则按x坐标)非降序排序
int cmp(const void *a,const void *b)
{
    if((*(edge *)a).w == (*(edge *)b).w)
    {
        return (*(edge *)a).x - (*(edge *)b).x;
    }
    return (*(edge *)a).w - (*(edge *)b).w;
}

//初始化集合
void Make_Set(int x)
{
    father[x] = x;
    rank[x] = 1;
}

//查找x元素所在的集合,回溯是压缩路径
int Find_Set(int x)
{
    if(x != father[x])
    {
        father[x] = Find_Set(father[x]);//递归压缩
    }
    return father[x];
}

//合并x,y所在的集合
void Union(int x,int y,int w)
{
    if(x == y) return;
    //将高度较小的连接在高度较大的树后
    if(rank[x] == rank[y]){
        rank[x] = rank[y] + 1;
        father[y] = x;
    }else if(rank[x] < rank[y])
        father[x] = y;
    else
        father[y] = x;
    sum += w;
}

//主函数
int main()
{
    int i,n;
    int x,y;
    char chx,chy;
    //读取边的数目
    cin>>n;

    //读取边信息并初始化集合
    for(i = 0; i < n; i++)
    {
        cin>>chx>>chy>>e[i].w;
        e[i].x = chx - 'A';
        e[i].y = chy - 'A';
        Make_Set(i);
    }
    //将边排序
    qsort(e,n,sizeof(edge),cmp);
    sum = 0;
    for(i = 0; i < n; i++)
    {
        x = Find_Set(e[i].x);
        y = Find_Set(e[i].y);
        if(x != y)
        {
            printf("%c - %c:%d\n",e[i].x + 'A',e[i].y + 'A',e[i].w);
            Union(x,y,e[i].w);
        }
    }
    printf("MST Total:%d\n",sum);
    return 0;
}

运行结果:

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值