sdut oj3362 数据结构实验之图论六：村村通公路(最小生成树)

题目链接：点击打开链接

数据结构实验之图论六：村村通公路

Time Limit: 1000MS Memory limit: 65536K

题目描述

当前农村公路建设正如火如荼的展开，某乡镇政府决定实现村村通公路，工程师现有各个村落之间的原始道路统计数据表，表中列出了各村之间可以建设公路的若干条道路的成本，你的任务是根据给出的数据表，求使得每个村都有公路连通所需要的最低成本。

输入

连续多组数据输入，每组数据包括村落数目N(N <= 1000)和可供选择的道路数目M(M <= 3000)，随后M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个村庄的编号和修建该道路的预算成本，村庄从1～N编号。 

输出

输出使每个村庄都有公路连通所需要的最低成本，如果输入数据不能使所有村庄畅通，则输出-1，表示有些村庄之间没有路连通。 

示例输入

5 8
1 2 12
1 3 9
1 4 11
1 5 3
2 3 6
2 4 9
3 4 4
4 5 6

示例输出

19

提示

 

代码实现：

#include <bits/stdc++.h>
#include <algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;

int cnt,flag;
int mp[1100][1100];
int vis[1100];
int dis[1100];
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] && min > dis[j]){
                min = dis[j];
                k = j;
            }
        }
        if(min == INF){
            flag = 1;
            break;
        }
        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;
        flag = 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();
        if(!flag)
            printf("%d\n",cnt);
        else
            printf("-1\n");
    }
    return 0;
}