Prim算法求最小生成树

Prim求最小生成树，使用邻接矩阵来存储每条路径，适用于稠密图

分析：先确定一个集合，每次在除去这个集合中的点里边确定一个距离集合最近的点加入集合，如果最近的一个点都无法到达，那么这个图就是非连通图，即无最小生成树，否则记录下这个距离。然后用记录的这个点去更新其余点到集合的距离。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=510,M=100010;
int g[N][N];
bool st[N];
int d[N];
int n,m;

int prim()
{
    int res=0;
    memset(d,0x3f,sizeof d);
    d[1]=0;//先把点1放到集合里边
    for(int i=0;i<n;i++)//每次循环确定一个点，注意这里不能写i<n-1，虽然确定了点1，但还需要用点1去更新其他点的距离
    {
        int t=-1;
        for(int j=1;j<=n;j++)//每次选出不是集合中且距离集合最近的点
        {
            if(!st[j]&&(t==-1||d[j]<d[t]))
                t=j;
        }
        if(d[t]>=0x3f3f3f3f)    return -999;//如果这个点不连通，则最小生成树不存在
        res+=d[t];//记录这个点到集合的距离
        for(int j=1;j<=n;j++)//用这个点去更新其他点到集合的距离
            if(!st[j]) 
                d[j]=min(d[j],g[t][j]);
        st[t]=true;//把这个点记录到集合里边
    }
    return res;
}

int main(void)
{
    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 res=prim();
    if(res==-999)   puts("impossible");
    else    printf("%d",res);
    return 0;
}