最小生成树

最小生成树

一、Prim求最小生成树

Prim最小生成树的原理是将一 点与集合中距离最近的点连接,在将其加入集合中,循环此操作。

代码与dijkstrs算法类似,如下:

const int INF = 0x3f3f3f3f;
int dist[N],g[N][N];
bool vis[N];

int prim()
{
    memset(dist,0x3f,sizeof dist);//dist储存的是每个点到集合的距离,先将其初始化为无限大
    int ans = 0;//存储答案
    for(int i = 0;i < n;i++)
    {
        int t = -1;
        for(int j = 1;j <= n;j++)//找出距离集合最小的点 
            if(!vis[j] && (t == -1 || dist[t] > dist[j]))
                t = j;
        
        if(i && dist[t] == INF)return INF;//如果不是第一个点,且距离集合的距离为无限大,说明不存在生成树
        
        if(i)ans += dist[t];
        
        for(int j = 1;j <= n;j++)dist[j] = min(dist[j],g[t][j]);//更新每个点到集合的距离
    }
}

二、Kruskal求最小生成树

Kruskal算法结合了并查集的原理,按边权从小到大排列,然后进行连接并计算边权和,以求最小生成树。

首选是利用类或者结构体储存参数,并重构<运算符以使用sort函数:

class Edge
{
publice:
    int a,b,c;
    bool operator<(const Edge &op)const
    {
        return c < op.c;
    }
}edge[N]; 

在将数据读入之后,利用sort函数将边权从小到大进行排序,最后就是调用Kruckal函数了:

int kruskal()
{
    for(int i = 1;i <= n;i++) f[i] = i;//并查集
    
    sort(edge,edge + m);
    
    itn ans = 0,cnt = 0;
    for(int i = 0;i < m;i++)
    {
        int a = edge[i].a,b = edge[i].b,c = edge[i].c;
        a = find(a),b = find(b);
        if(a != b)
        {
            f[a] = b;
            ans += c;
            cnt++;
        }
    }
    
    if(cnt < n - 1)return INF;//定义常量INF = 0x3f3f3f3f
    return ans;
}

最后是find函数,具体的可参考这篇文章:并查集的原理与应用_Kicamon的博客-CSDN博客

int find(int a)
{
    if(f[a] != a) f[a] = find(f[a]);
    return f[a];
} 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值