Prim算法求最小生成树 Prim模板

题目：

题解：

找到集合外距离集合最近的点，如果找不到则没有最小生成树（第一个点除外），在路径上加上这个点的距离（第一个点除外），然后用这个点更新更新其他点到集合的距离，在给这个点打上在集合的标记。

代码：

#include<bits/stdc++.h>
using namespace std;

const int N=510;

int g[N][N];
int d[N][N];
int dist[N];
bool st[N];
int n,m;

int prim(){
    int res=0;
    memset(dist,0x3f,sizeof dist);
    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;
        }
        st[t]=true;
        if(i&&dist[t]==0x3f3f3f3f)return 0x3f3f3f3f;
        //必须先加上这一条边，再更新其他边，否则有自环时可能会出错
        if(i)res+=dist[t];
        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);
    for(int i=0;i<m;i++){
        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==0x3f3f3f3f)printf("impossible\n");
    else printf("%d\n",t);
    return 0;
}