prim算法

#include<iostream>
#include<cstring>

using namespace std;

const int N=510,INF=0x3f3f3f3f;

int g[N][N];
int dist[N];
int n,m;
bool mark[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++){//遍历dist找出不在最小生成树集合中的点的最短边
            if(!mark[j]&&(t==-1||dist[j]<dist[t]))t=j;
        }

        if(i&&dist[t]==INF)return INF;//如果不是第一个加入的点并且不在图中的点没有符合要求的边
        if(i)res+=dist[t];

        for(int j=1;j<=n;j++)dist[j]=min(dist[j],g[t][j]);//更新dist数组

        mark[t]=1;//将该点加入最小生成树集合中
    }

    return res;//返回最小生成树总的权值


}

int main(){

    memset(g,0x3f,sizeof g);
    cin>>n>>m;
    
    int v1,v2,c;
    while(m--){
        cin>>v1>>v2>>c;
        g[v1][v2]=g[v2][v1]=min(c,g[v1][v2]);//用min是为了防止重边，有重边时取最小的
    }

    int ans=prim();

    if(ans==INF)cout<<"impossible";
    else cout<<ans;
    return 0;
}

1）memset可以初始化二维数组
2）0x3f和0x3f3f3f3f不是同一个数