08-图7 公路村村通 (30 分)
现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。
输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。
输入样例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3结尾无空行
输出样例:
12
结尾无空行
代码:
#include <bits/stdc++.h>
using namespace std;
const int MAX_SIZE = 3010;
int f[MAX_SIZE];
int n, m;
struct Edge
{
int u, v, w;
} edge[MAX_SIZE];
bool cmp(Edge a, Edge b) { return a.w < b.w; }
void init()
{
for (int i = 1; i < MAX_SIZE; i++)
f[i] = i;
}
int find(int i)
{
if (f[i] != i)
f[i] = find(f[i]);
return f[i];
}
int kruskal()
{
int num = 0;
int ans = 0;
init();
sort(edge + 1, edge + 1 + m, cmp);
for (int i = 1; i <= m; i++)
{
int t1 = find(edge[i].u);
int t2 = find(edge[i].v);
if (t1 == t2)
continue;
f[t1] = t2;
num++;
ans += edge[i].w;
}
if (num == n - 1)
return ans;
else
return -1;
}
int main()
{
cin >> n >> m;
for (int i = 1; i <= m; i++)
{
scanf("%d %d %d", &edge[i].u, &edge[i].v, &edge[i].w);
}
printf("%d", kruskal());
return 0;
}