这题和村村通公路是一样的,详解请参考村村通公路
数据结构实验之图论九:最小生成树
Time Limit: 1000MS
Memory Limit: 65536KB
Problem Description
有n个城市,其中有些城市之间可以修建公路,修建不同的公路费用是不同的。现在我们想知道,最少花多少钱修公路可以将所有的城市连在一起,使在任意一城市出发,可以到达其他任意的城市。
Input
输入包含多组数据,格式如下。
第一行包括两个整数n m,代表城市个数和可以修建的公路个数。(n <= 100, m <=10000)
剩下m行每行3个正整数a b c,代表城市a 和城市b之间可以修建一条公路,代价为c。
Output
每组输出占一行,仅输出最小花费。
Example Input
3 2 1 2 1 1 3 1 1 0
Example Output
2 0
#include <bits/stdc++.h> #define INF 0x3f3f3f3f using namespace std; int a[1100][1100], vis[1100], dis[1100]; int n, m; void prim() { int d = 1; int sum = 0; int pos; for(int i = 1; i <= n; i++) { dis[i] = a[1][i]; } vis[1] = 1; for(int i = 1; i < n; i++) { int mm = INF; for(int j = 1; j <= n; j++) { if(vis[j] == 0 && dis[j] < mm) { pos = j; mm = dis[j]; } } if(mm == INF) { d = 0; break; } sum += mm; vis[pos] = 1; for(int i = 1; i <= n; i++) { if(vis[i] == 0 && dis[i] > a[pos][i]) { dis[i] = a[pos][i]; } } } if(d == 0) { cout << "-1" << endl; } else { cout << sum << endl; } } int main() { while(cin >> n >> m) { memset(a, INF, sizeof(a)); memset(vis, 0, sizeof(vis)); memset(dis, 0, sizeof(dis)); int i, j, k; while(m--) { cin >> i >> j >> k; if(a[i][j] > k) { a[i][j] = a[j][i] = k; } } prim(); } return 0; }