布线问题
-
描述
-
南阳理工学院要进行用电线路改造,现在校长要求设计师设计出一种布线方式,该布线方式需要满足以下条件:
1、把所有的楼都供上电。
2、所用电线花费最少
-
输入
-
第一行是一个整数n表示有n组测试数据。(n<5)
每组测试数据的第一行是两个整数v,e.
v表示学校里楼的总个数(v<=500)
随后的e行里,每行有三个整数a,b,c表示a与b之间如果建铺设线路花费为c(c<=100)。(哪两栋楼间如果没有指明花费,则表示这两栋楼直接连通需要费用太大或者不可能连通)
随后的1行里,有v个整数,其中第i个数表示从第i号楼接线到外界供电设施所需要的费用。( 0<e<v*(v-1)/2 )
(楼的编号从1开始),由于安全问题,只能选择一个楼连接到外界供电设备。
数据保证至少存在一种方案满足要求。
输出
- 每组测试数据输出一个正整数,表示铺设满足校长要求的线路的最小花费。 样例输入
-
1 4 6 1 2 10 2 3 10 3 1 10 1 4 1 2 4 1 3 4 1 1 3 5 6
样例输出
-
4
-
第一行是一个整数n表示有n组测试数据。(n<5)
思路:先将花费从小到大排序,然后判断a、b两号楼之间有没有连通,如果没有,则将他们归到一个集合中去,并加上花费
注:判断
a、b两号楼之间有没有连通,不能只查f[a]==f[b];而要通过find函数的返回值,如果t1!=t2,则说明两数不属于一个集合,此时应把花费加上
#include<stdio.h>
#include<stack>
#include<math.h>
#include<queue>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
int f[1005],sum=0;
struct node
{
int a,b,c;
} s[150000];
int cmp(node A,node B)
{
return A.c<B.c;
}
int find(int x)
{
return f[x]==x?x:find(f[x]);
}
int main()
{
int T,v,e,i,x,min;
scanf("%d",&T);
while(T--)
{
sum=0;min=9999999;
scanf("%d%d",&v,&e);
for(i=1; i<=v; i++)
f[i]=i;
for(i=0; i<e; i++)
scanf("%d%d%d",&s[i].a,&s[i].b,&s[i].c);
sort(s,s+e,cmp);
for(i=0; i<e; i++)
{
if(f[s[i].a]!=f[s[i].b])
{
int t1=find(s[i].a);
int t2=find(s[i].b);
if(t1!=t2)
{
sum+=s[i].c;
f[t2]=t1;
}
}
}
for(i=0;i<v;i++)
{
scanf("%d",&x);
if(x<min)
{
min=x;
}
}
printf("%d\n",sum+min);
}
return 0;
}