最小生成树--Prim

Posted 2020-10-30 anhuixuyin218

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<algorithm>
#include<string>
#include<stack>
#include<queue>
#include<map>
#define MAX 510
const int INF=1000000000;
using namespace std;
//顶点到集合s的最短距离
int d[MAX],G[MAX][MAX];
int n,m;
bool isVisit[MAX]={false};
//返回最小生成树的边权之和
int prim(){
    fill(d,d+MAX,INF);
    d[0]=0;
    int ans=0;
    for(int i=0;i<n;i++){
        int u=-1,MIN=INF;
        for(int j=0;j<n;j++){
            if(isVisit[j]==false&&d[j]<MIN){
                u=j;
                MIN=d[j];
            }
        }
        if(u==-1) return -1;
        isVisit[u]=true;
        ans+=d[u];
        for(int v=0;v<n;v++){
            if(isVisit[v]==false&&G[u][v]!=INF&&G[u][v]<d[v])
                d[v]=G[u][v];
        }
    }
    return ans;
}

int main(){
    int u,v,w;
    //顶点个数，边数
    scanf("%d%d",&n,&m);
    //初始化图
    fill(G[0],G[0]+MAX*MAX,INF);
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&u,&v,&w);
        G[u][v]=G[v][u]=w;
    }
    int ans=prim();
    printf("%d\n",ans);
    return 0;
}

　　类似Dijkstra算法，但是此时d[]表示顶点Vi与集合S的最短距离