题目链接
现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数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;
int f[1001];
typedef struct Node{
int aa;
int bb;
int fei;
}Node;
bool cmp(Node a,Node b){
return a.fei < b.fei;
}
int getf(int x){
if(f[x] == x)
return f[x];
f[x] = getf(f[x]);
return f[x];
}
int merge(int x,int y){
int tx = getf(x);
int ty = getf(y);
if(tx != ty){
f[ty] = tx;
return 1;
}
return 0;
}
void init(int n){
for(int i = 1;i <= n;i++)
f[i] = i;
return;
}
int main(){
int n,m,a,b,fee,sum,cnt;
Node p[3001];//设置为3001,一开始设置成了1001,后来发现题目中候选道路数目M≤3N,与m看齐
scanf("%d%d",&n,&m);
init(n);//并查集初始化
for(int i = 0;i < m;i++){
scanf("%d%d%d",&a,&b,&fee);
p[i].aa = a;
p[i].bb = b;
p[i].fei = fee;
}
sort(p,p + m,cmp); //按照权值的大小进行排序
sum = 0,cnt = 0;
for(int i = 0;i < m;i++)//最小生成树,循环n-1次,结点的个数减一
if(merge(p[i].aa,p[i].bb)){
sum += p[i].fei;
cnt++;//统计边数
}
if(cnt == n- 1)
printf("%d\n",sum);
else
printf("-1\n");
return 0;
}