hihocoder1097---最小生成树一Prim算法

转载于 2018-08-09 16:37:00 发布 · 91 阅读

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/SUMaywlx/p/9449556.html

本文详细介绍了Prim算法用于求解最小生成树问题的过程与实现细节。通过划分顶点集合并不断选择最优边来逐步构建最小生成树，算法实现了高效的图遍历。文章提供了完整的C++代码示例，展示了如何初始化图结构、更新顶点距离以及最终计算最小生成树的总权重。

//与Dijsktra思想下相同，划分两个不同集合
//区别：d[]数组含义不同，此处是到集合的最短距离
//优化方式不同的，d[v]=G[u][v];
#include<bits/stdc++.h>
using namespace std;
const int M=1010;
const int inf=0x3fffffff;
int n;
int G[M][M];
int d[M];
bool vis[M];
int prim(){
    fill(d,d+M,inf);
    fill(vis,vis+M,0);
    d[0]=0;
    int ans=0;
    for(int i=0;i<n;i++){
        int u=-1;int Min=inf;
        for(int j=0;j<n;j++){
            if(vis[j]== 0 && d[j]<Min){
                u=j;Min=d[j];
            }
        }
        if(u == -1) return -1;
        vis[u]=1;
        ans+=d[u];
        for(int k=0;k<n;k++){
            if(!vis[k] && G[u][k]<d[k]){
                d[k]=G[u][k];
            }
        }
    }
    return ans;
}
int main(){
    while(scanf("%d",&n)!=EOF){
        for(int i=0;i<n;i++){
                for(int j=0;j<n;j++){
                    scanf("%d",&G[i][j]);
                }
        }
        int ans=prim();
        printf("%d\n",ans);
    }
    return 0;
}