7-10 公路村村通 (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 <iostream> #include <algorithm> //感觉这道题并查集没什么用 using namespace std; struct Node{ int s, e;//s为起始点,e为终点 int cost;//路程花费 }; struct Node edge[3001]; //边数 int a[1001];//村落数 (并查集容器) int ans = 0;//记录边数 bool cmp(Node x, Node y){//按从小到大排序 return x.cost < y.cost; } int FindBoot(int i){//找根结点+路径压缩 if(i != a[i]){//如果自身不是自身的根 a[i] = FindBoot(a[i]);//向上查找 } return a[i];//路径压缩 } int main(){ bool flag = false; int n, m; int sum = 0; cin >> n >> m; for(int i = 1; i <= n; i++){ a[i] = i;//自己是自己的根 } for(int i = 0; i < m; i++){ cin >> edge[i].s >> edge[i].e >> edge[i].cost; } sort(edge, edge + m, cmp); for(int i = 0; i < m; i++){ int t1 = FindBoot(edge[i].s); int t2 = FindBoot(edge[i].e); if(t1 != t2){ ans++;//边数+1 a[t1] = t2;//把集合1并入集合2中 sum += edge[i].cost;//把花费加进来 if(ans >= n - 1){ flag = true; break; } } } if(!flag){ cout << "-1" << endl; }else{ cout << sum << endl; } return 0; }