数据结构实验之图论九：最小生成树

数据结构实验之图论九：最小生成树

Time Limit: 1000 ms Memory Limit: 65536 KiB

Submit Statistic

Problem Description

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

 

Input

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

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

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

 

Output

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

Sample Input

3 2
1 2 1
1 3 1
1 0

Sample Output

2
0

Hint

Source

赵利强

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <vector>
#define inf 0x3fffff
using namespace std;
int gra[3001][3001];
int vis[1001], dist[1001];

void prim(int n) {
    for (int i = 1; i <= n; i++) dist[i] = gra[1][i];
    int flag = 1;
    for (int i = 1; i < n; i++) {
        int min = inf;
        int index;
        for (int j = 1; j <= n; j++) {
            if (!vis[j] && dist[j] < min) {
                min = dist[j];
                index = j;
            }
        }
        // 找不到 相通
        if (min == inf) {
            flag = 0;
            break;
        }
        vis[index] = 1;
        for (int j = 1; j <= n; j++) {
            if (!vis[j] && gra[index][j] < dist[j]) dist[j] = gra[index][j];
        }
    }
    if (flag) {
        int ans = 0;
        for (int i = 1; i <= n; i++) ans += dist[i];
        cout << ans << endl;
    } else {
        cout << "0" << endl;
    }
}

int main() {
    int n, m, u, v, w;
    while (cin >> n >> m) {
        memset(vis, 0, sizeof(vis));
        for (int i = 0; i <= n; i++) {
            for (int j = 0; j <= n; j++) {
                if (i == j)
                    gra[i][j] = 0;
                else
                    gra[i][j] = inf;
            }
        }
        for (int i = 0; i < m; i++) {
            cin >> u >> v >> w;
            if (w < gra[u][v]) gra[u][v] = gra[v][u] = w;
        }
        if (m == 0)
            printf("0\n");
        else
            prim(n);
    }
}