问题 G: 建设电力系统
题目描述
小明所在的城市由于下暴雪的原因,电力系统严重受损。许多电力线路被破坏,因此许多村庄与主电网失去了联系。政府想尽快重建电力系统,所以,身为程序员的你被赋予了一项任务,就是编程计算重建电力系统的最少花费,重建的电力系统必须保证任意两个村庄之间至少存在一条通路。
输入
输入的第一行为一个整数T(1<=T<=50),表示有T组测试数据。
每组输入第一行是两个正整数N,E(2<=N<=500,N<=E<=N*(N-1)/2),分别表示村庄的个数和原有电力线路的个数。
接下来的E行,每行包含三个整数A,B,K(0<=A,B<N,0<=K<1000)。A和B分别表示电力线路的起始村庄代号。如果K=0,表示这条线路仍然正常。如果K是一个正整数,表示这条线路需要花费K的代价来重建。
题目保证输入中没有重边,也没有起始村庄相同的边。
每组输入第一行是两个正整数N,E(2<=N<=500,N<=E<=N*(N-1)/2),分别表示村庄的个数和原有电力线路的个数。
接下来的E行,每行包含三个整数A,B,K(0<=A,B<N,0<=K<1000)。A和B分别表示电力线路的起始村庄代号。如果K=0,表示这条线路仍然正常。如果K是一个正整数,表示这条线路需要花费K的代价来重建。
题目保证输入中没有重边,也没有起始村庄相同的边。
输出
对于每组输入,输出重建电力系统所需的最小花费,以此来保证任意两个村庄之间至少存在一条通路。
样例输入
1
3 3
0 1 5
0 2 0
1 2 9
样例输出
5
#include<stdio.h>
#include<algorithm>
using namespace std;
#define N 510
int f[N];
struct edge{
int u;
int v;
int w;
};
struct edge e[N*N];
int cmp(struct edge a,struct edge b)
{
return a.w < b.w ;
}
int find(int v)
{
if(f[v] == v)
return v;
else
{
f[v] = find(f[v]);
return f[v];
}
}
int merge(int u,int v)
{
int t1,t2;
t1 = find(u);
t2 = find(v);
if(t1 != t2)
{
f[t1] = t2;
return 1;
}
return 0;
}
int main()
{
int t;
int n,m;
int i;
int sum,count;
scanf("%d",&t);
while( t --)
{
count = 0;
sum = 0;
scanf("%d%d",&n,&m);
for( i = 0; i < m; i ++)
scanf("%d%d%d",&e[i].u ,&e[i].v ,&e[i].w );
for( i = 0; i < n; i ++)
f[i] = i;
sort(e,e+m,cmp);
for( i = 0; i < m; i ++)
{
if(merge(e[i].u ,e[i].v))
{
sum += e[i].w ;
count2++;
}
if(count2 == n-1)
break;
}
printf("%d\n",sum);
}
return 0;
}