acwing 858 Prim算法求最小生成树

NirvanaCensor

已于 2023-02-06 22:54:06 修改

阅读量164

点赞数

分类专栏： acwing刷题笔记文章标签：算法 Powered by 金山文档

于 2023-02-06 22:48:06 首次发布

本文链接：https://blog.csdn.net/NirvanaCensor/article/details/128909681

版权

acwing刷题笔记专栏收录该内容

21 篇文章 0 订阅

订阅专栏

该代码示例展示了如何使用Prim算法在图中找到连接所有节点的最小生成树。算法从一个初始节点开始，逐步添加与当前生成树具有最短边长的未连接节点，直到所有节点都被包含或发现孤立点。程序读取图的边权重，处理可能存在的重边，并返回最小生成树的总权重。如果图不完全连通，返回inf表示不可能。

摘要由CSDN通过智能技术生成

思路：现规定一个集合，集合初始包含第一个点，主题流程是在图中找出最短的与集合相连的边，将这条边的另一个顶点加入集合，然后根据新加入的这个点更新其他的点的边，如此重复，直到所有点加入集合或者找出孤立点/无法连通的部分。得到的结果就是最小生成树，prim算法思想的核心是找到不在集合内的连接集合的最短边

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 510,INF = 0x3f3f3f3f;

int n,m;
int g[N][N],dist[N];//dist保存的是当前点到生成树（已有集合）的距离
bool st[N];//判断是否在生成树（集合）内

int prim()
{
    memset(dist,0x3f,sizeof dist);
    int res = 0;
    for(int i = 0;i < n;i++)
    {
        int t = -1;//t用与保存距离已有集合最近的未加入集合的点
        for(int j = 1;j <= n;j++)
        {
            if(!st[j] && (t == -1|| dist[t] > dist[j]))//不在集合内且t到集合的距离大于j到集合的距离
                t = j;
        }
        
        if(i && dist[t] == INF) return INF;//不是第一个点，且该图不完全连通（存在孤立点）
        if(i) res += dist[t];//点加入集合，边长进入累计
        
        st[t] = true;//把加入的点置 1
        
        for(int j = 1;j <= n;j++)//用新加入的点更新所有点到集合的距离
            dist[j] = min(dist[j],g[t][j]);
    }
    return res;
}

int main()
{
    scanf("%d%d",&n,&m);
    
    memset(g,0x3f,sizeof g);
    
    while(m--)
    {
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        g[a][b] = g[b][a] = min(g[a][b] , w);//存在重边保存最小边
    }
    int t = prim();
    
    if(t == INF) puts("impossible");
    else cout << t << endl;
    
    return 0;
}