数据结构实验之图论九:最小生成树
Time Limit: 1000 ms Memory Limit: 65536 KiB
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);
}
}