最小生成树_Prim算法

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
//prim算法与Dijkstra很像，区别是他需要的更新的是到集合的距离
//稠密图，无向图：邻接矩阵存两遍
const int N=510;

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

int prim()
{
    memset(dist, 0x3f, sizeof dist);
    dist[1]=0;
    
    for(int i=1;i<=n;i++)
    {
        int t=-1;
        
        //跟dijkstra相同的步骤先通过优先队列找距离最近的点（问题：同样的算法既可以找点的最近也可以找集合的？ 解答：可以，因为我在后面更新距离的时候更新的就是点到集合的距离，所以下一个优先点就是距离集合最近的点）
        for(int j=1;j<=n;j++)
            if(!st[j] && (t==-1 || dist[t]>dist[j]) )
                t=j;
                
        if(dist[t]==0x3f3f3f3f) return -1;//因为下面的循环会更新此时集合所能接触到点的距离，因此如果距离仍未无穷，说明这个点是不连接集合的，那么就是没有最小生成树，直接退出即可
        
        st[t]=true;//说明已经走过了
        sum+=dist[t];//用来记录此时最小生成树的距离
        
        for(int j=1;j<=n;j++)
            dist[j]=min(dist[j], g[t][j]);//这里的更新循环与Dijkstra的区别在于这里是更新与集合的最近距离，所以不需要再加上dist[t](t到起始点的距离)
    }
    
    return sum;
    
}


int main()
{
    cin>>n>>m;
    
    memset(g, 0x3f, sizeof g);
    
    while(m--)
    {
        int a,b,w;
        cin>>a>>b>>w;
        if(a!=b) g[b][a]=g[a][b]=min(g[a][b],w);//关键点①：必须这么写邻接矩阵，否则会被自环搞的无解
    }
    
    int t=prim();
    if(t==-1) cout<<"impossible"<<endl;
    else cout<<t<<endl;
    
    return 0;
    
}