最小生成树 prim算法 C++实现

最新推荐文章于 2022-12-21 11:10:56 发布

areen7003

最新推荐文章于 2022-12-21 11:10:56 发布

阅读量478

点赞数

文章标签： c/c++ 数据结构与算法

原文链接：http://www.cnblogs.com/xLester/p/7570293.html

版权

本文详细介绍了使用邻接矩阵实现的最小生成树算法，通过一个具体的C++代码示例，展示了如何从给定的节点和边中寻找最小生成树的过程。算法首先读取节点数和边数，然后构建邻接矩阵，接着通过Prim算法逐步选择未被包含在树中的最短边，最终输出最小生成树的总权重。

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

#include <iostream>
#include <vector>
#define Inf 0x7fffffff
using namespace std;
int main()
{
    int n,m;//节点个数和边数
    cin>>n>>m;//输入节点个数和边数;
    vector<vector<int>> v(n+1,vector<int>(n+1,Inf));//邻接矩阵
    for(int i=0;i<m;++i)
    {
        int s,e,w;
        cin>>s>>e>>w;
        v[s][e]=v[e][s]=w;
    }
    int s=1,sumcost=0;//开始收录的节点,累计最小花费
    vector<int> cost=v[s];
    vector<bool> vis(n+1,false);//是否已经收录
    vis[s]=true;
    for(int i=1;i<n;++i)
    {
        int mincost=Inf,u=-1;
        for(int i=1;i<=n;++i) if(!vis[i]&&cost[i]<mincost) mincost=cost[u=i];//找到最小花费
        sumcost+=mincost;
        vis[u]=true;
        for(int i=1;i<=n;++i) if(!vis[i]&&cost[i]>v[i][u]) cost[i]=v[i][u];//通过新收录的节点u，重新更新最小花费
    }
    cout<<sumcost;
    return 0;
}