858. Prim算法求最小生成树

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

//prim算法 每次放入不在连通块中 最近的一个点 连通块加上距离 然后更新迭代其余所有点
#include<iostream>
#include<cstring>
using namespace std;

const int N=510,INF=0x3f3f3f3f;
int g[N][N];//g[a][b] a->b的距离
int dist[N];//dist[i]:i到连通块中的最小距离
bool st[N];//是否在连通块中
int n,m;

int prim()
{
    memset(dist,0x3f,sizeof dist);
    int res=0;//连通块长度
    
    for(int i=0;i<n;i++)//n个点 循环n次 Dijkstra可循环n-1次(起点直接放入)
    {
        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;//不是起点 但是到连通块中点的距离为INF
        if(i) res+=dist[t];//不是起点 距离加上
        //这步要放在更新其余点之前，因为如果有自环，g[t][t]可能会更新dist[t]
        for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]);
        
        st[t]=true;//放入连通块
    }
    
    return res;
}

int main()
{
    scanf("%d%d",&n,&m);
    memset(g,0x3f,sizeof g);
    
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        g[a][b]=g[b][a]=min(g[a][b],c);
    }
    
    int t=prim();
    if(t==INF) printf("impossible\n");
    else printf("%d\n",t);
    
    return 0;
}