现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。
输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。
思路: 将两节点与花费的结构体按花费升序排序,遍历每个结构体,若不互相连通,将两点加入并查集,并更新花费。最后遍历任意两点判断是否存在不通畅的情况
代码:
#include<iostream>
#include<queue>
#include<stack>
#include<cstring>
#include<stdlib.h>
#include<cstdio>
#include<iostream>
#include<algorithm>
#define LL long long
using namespace std;
#define MAXN 3001
struct node{
int f, s, cost;
};
bool cmp(node a, node b){
return a.cost < b.cost;
}
node c[MAXN];
int F[MAXN];
void init(int n){
for(int i = 1; i <= n; i++)
F[i] = i;
}
int find(int x){
if(F[x] != x)
F[x] = find(F[x]);
return F[x];
}
bool same(int x, int y){
if(find(x) == find(y))
return true;
return false;
}
void unionn(int x, int y){
int t1 = find(x);
int t2 = find(y);
if(t1 != t2){
F[t1] = t2;
}
}
//int w[MAXN][MAXN];
int main(){
ios :: sync_with_stdio(false);
int N, M;
cin >> N >> M;
init(N);
int sum = 0;
for(int i = 0; i < M; i++){
cin >> c[i].f >> c[i].s >> c[i].cost;
//w[c[i].f][c[i].s] = c[i].cost;
//unionn(c[i].f, c[i].s);
}
sort(c, c+M, cmp);
for(int i = 0; i < M; i++){
if(!same(c[i].f, c[i].s)){
sum += c[i].cost;
unionn(c[i].f, c[i].s);
}
}
for(int i = 1; i <= N; i++){
for(int j = i+1; j <= N; j++){
if(!same(i, j)){
cout << -1 << endl;
return 0;
}
}
}
cout << sum << endl;
//system("pause");
return 0;
}