AcWing 858. Prim算法求最小生成树

朴素Prim算法求最小生成树

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

AcWing 858. Prim算法求最小生成树：图解+详细代码注释（带上了保存路径） - AcWing

#include<iostream>
#include<cstring>
using namespace std;
const int N = 510,INF = 0x3f3f3f3f;
int n,m;
int g[N][N];
int 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;
        for(int j = 1;j<=n;j++)
        {
            if(!st[j]&&(t==-1||dist[t]>dist[j]))
            {
                t = j;
            }
        }
        
        if(i&&dist[t] == INF) return INF;
        //为了防止t点自己有负权边，会自己更新自己
        //因此我们先将res进行更新,再用t点去更新其他点的dist
        //并且注意，第一个点加进去的时候，这个时候还没有边，千万不要更新res
        //不然res会在第一个点的时候就更新成INF了
        if(i) res += dist[t];
        st[t] = true;
        
        //更新其他点的dist
        for(int j = 1;j<=n;j++) dist[j] = min(dist[j],g[t][j]);
    }
    return res;
}
int main()
{
    cin>>n>>m;
    memset(g,0x3f,sizeof g);
    while(m--)
    {
        int a,b,c;
        cin>>a>>b>>c;
        g[a][b] = g[b][a] = min(g[a][b],c);
    }
    int t = prim();//返回值为最小生成树中所有边权数之和
    
    if(t == INF)puts("impossible");
    else cout<<t<<endl;
    return 0;
}