2016夏季练习——最小生成树

最新推荐文章于 2020-10-27 21:22:41 发布

kissacm

最新推荐文章于 2020-10-27 21:22:41 发布

阅读量234

点赞数 1

分类专栏： ACM算法

本文链接：https://blog.csdn.net/kissacm/article/details/51993931

版权

ACM算法专栏收录该内容

179 篇文章 0 订阅

订阅专栏

来源：POJ1287

prim算法，还是比较轻松的

解释在注释里面

代码：

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
int n,R;
const int INF = 0x3f3f3f3f;
const int MAXN = 55;
int mp[MAXN][MAXN];
int d[MAXN];
bool vis[MAXN];
int ans;
int prim(){
    int pos,dm,ans=0;
    //initial
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=n;i++){
        d[i]=mp[1][i];

    }
    //过程：
    for(int i=1;i<=n;i++){
        dm=INF;//一开始去最大值
        pos=-1;//记录下一个加入的节点的ID
        for(int j=1;j<=n;j++){//找下一个没有访问过，但是距离我最短的点
            if(!vis[j]&&dm>d[j]){
                dm=d[j];
                pos=j;
            }
        }
        if(dm==INF) return -1;//如果没找到，就说明图是非联通的
        ans+=dm;//加入
        vis[pos]=true;//加入这个点
        for(int j=1;j<=n;j++){//跟新我已有的集合都各个点的最短距离
            if(!vis[j]&&d[j]>mp[pos][j]){
                d[j]=mp[pos][j];
            }
        }
    }//当n个点都加入的时候，循环结束，算法结束
    return ans;
}
int main(){
	while(scanf("%d",&n)!=EOF){
        if(!n) break;
        scanf("%d",&R);
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(i==j) mp[i][j]=0;
                else mp[i][j]=INF;
            }
        }
        int a,b,c;
        for(int i=0;i<R;i++){
            scanf("%d%d%d",&a,&b,&c);
            if(mp[a][b]>c) mp[a][b]=mp[b][a]=c;
        }
        //以上输入的处理都是套路，不解释
        cout<<prim()<<endl;;
	}
	return 0;
}