Prim算法求最小生成树问题

什么是Prim算法？

prim 算法干的事情是：给定一个无向图，在图中选择若干条边把图的所有节点连起来。要求边长之和最小。在图论中，叫做求最小生成树。

prim 算法采用的是一种贪心的策略。

每次将离连通部分的最近的点和点对应的边加入的连通部分，连通部分逐渐扩大，最后将整个图连通起来，并且边长之和最小。

具体步骤

dist[i]   <- 正无穷

for n 次每次确定一个点到连通块的最短距离，，，与dijstra算法很相似，区别在prim中dist是距离原点的距离，dijstra中dist 是距离连通块的距离

{

 for 与当前点相连的所有点

       找到距离连通块最近的点t

           for 用t 来更新其他点到连通块的距离

}

题目链接858. Prim算法求最小生成树 - AcWing题库

c++代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 510;
int g[N][N];
int n,m,res;
bool st[N];//是否在连通块中
int dist[N];//到连通块的距离
int prim()
{
    memset(dist,0x3f,sizeof dist);//初始化到连通块距离为无穷
    dist[1] = 0;//第一个点到连通块距离为0，但是没有加入连通块，原因是如果st[1] = true,（1）处t 就不会为1，需要让t = 1来更新其他点
    for(int i = 0; i < n; i ++ )
    {
        int t = 0;
        for(int j = 1; j <= n; j ++ )
        {
            if(!st[j] && dist[j] < dist[t])//（1）
            t = j;
        }
        if(dist[t] == 0x3f3f3f3f) return 0x3f3f3f3f;
        st[t] = true;
        res += dist[t];
        for(int j = 1; j <= n; j ++ )//扩展
        dist[j] = min(dist[j],g[t][j]);
    }
    return res;
}
int main()
{
    memset(g,0x3f,sizeof g);//初始化，可以让没有边的点之间的距离为无穷大，看成没有边
    cin >> n >> m;
    for(int i = 0; i < m; i ++ )
    {
        int a,b,c;
        cin >> a >> b >> c;
        g[a][b] = g[b][a] = min(g[a][b],c);//无向边+解决重边问题，，为什么不解决自环问题
    }
        int t = prim();
        if(t == 0x3f3f3f3f)puts("impossible");
        else cout << t << endl;
        return 0;
}