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

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

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 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#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);
    }
}

 

你可能感兴趣的:(《数据结构》图论)