sdut oj2144 图结构练习——最小生成树(普利姆算法)

题目链接：点击打开链接

图结构练习——最小生成树

Time Limit: 1000MS Memory limit: 65536K

题目描述

 有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。

 

输入

 输入包含多组数据，格式如下。

第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)

剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。

 

输出

 每组输出占一行，仅输出最小花费。

示例输入

3 2
1 2 1
1 3 1
1 0

示例输出

2
0

提示

 

代码实现：

#include <bits/stdc++.h>
#include <algorithm>
using namespace std;
const int INF = 1e9 + 7;

int cnt;
int mp[110][110];
int vis[110];
int dis[110];
int vexnum,arcnum;

void prim(){
    int k;
    vis[1] = 1;
    for(int i = 1;i <= vexnum;i++)
        dis[i] = mp[1][i];///初始化
    for(int i = 2;i <= vexnum;i++){
        int min = INF;
        for(int j = 1;j <= vexnum;j++){///找到权值最小的
            if(!vis[j] && dis[j] < min){
                min = dis[j];
                k = j;
            }
        }
        vis[k] = 1;
        cnt += min;
        for(int j = 1;j <= vexnum;j++){///更新他们的权值
            if(!vis[j] && dis[j] > mp[k][j])
            dis[j] = mp[k][j];
        }
    }
}

int main(){
    int u,v,w;
    while(~scanf("%d%d",&vexnum,&arcnum)){
        memset(mp,INF,sizeof(mp));
        memset(vis,0,sizeof(vis));
        cnt = 0;
        for(int i = 0;i < arcnum;i++){
            scanf("%d%d%d",&u,&v,&w);
            if(mp[u][v] > w)
                mp[u][v] = mp[v][u] = w;
        }
        prim();
        printf("%d\n",cnt);
    }
    return 0;
}