7-7 公路村村通
现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数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
解
这道题,嘶,最小生成树问题,对应的有两种算法 : Prim(普里姆)算法
和 Kruskal(克鲁斯卡尔)算法
这道题两种算法都能解,我用的后者 Kruskal(克鲁斯卡尔)算法
#include<bits/stdc++.h>
using namespace std;
#define M 3003
struct s{
int x, y;
int weight;
}t[M];
int parent[M];
bool cmp(s &a, s &b)
{
return a.weight < b.weight;
}
int findparent(int a)
{
while(a != parent[a])
a = parent[a];
return a;
}
int Kruskal(int n, int m)
{
int num = 0, sum = 0;
for(int i = 0; i < m; i++)
{
int a = findparent(t[i].x);
int b = findparent(t[i].y);
if(a != b) // 若 a != b 则说明目前生成树还没有生成环路
{
parent[a] = b; /// 表示此顶点已在生成树集合中
num++;
sum += t[i].weight;
}
if(num == n - 1)
break;
}
if(num == n - 1)
return sum;
return -1;
}
int main()
{
int n, m;
cin >> n >> m;
for(int i = 0; i < n; i++)
parent[i] = i;
for(int i = 0; i < m; i++)
{
cin >> t[i].x >> t[i].y >> t[i].weight;
}
sort(t, t + m, cmp);
cout << Kruskal(n, m) << endl;
return 0;
}