NYOJ 38 布线问题_(解法1 Kruskal算法)

时间限制：1000 ms  |  内存限制：65535 KB

难度：4

描述

南阳理工学院要进行用电线路改造，现在校长要求设计师设计出一种布线方式，该布线方式需要满足以下条件：
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开始），由于安全问题，只能选择一个楼连接到外界供电设备。
数据保证至少存在一种方案满足要求。

输出

每组测试数据输出一个正整数,表示铺设满足校长要求的线路的最小花费。

样例输入

14 61 2 102 3 103 1 101 4 12 4 13 4 11 3 5 6

样例输出

4

代码如下

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int MAXN = 505 ;struct ArcNode{int v1,v2;//v1、v2表示可连通的楼int cost;//cost表示连通v1、v2的花费};int father[MAXN],add[MAXN];int v,e,s;bool cmp(const ArcNode &lhs, const ArcNode &rhs){return lhs.cost < rhs.cost;}void Kruskal(ArcNode *node){int i,j,k,x,y;i=j=0;s=0;while(j<v-1){x=father[node[i].v1-1];y=father[node[i].v2-1];if (x!=y){for(k=0;k<v;k++)if(father[k]==y)father[k]=x;s+=node[i].cost;++j;}++i;}}int main(){int n;int i;scanf("%d",&n);while(n--){scanf("%d%d",&v,&e);ArcNode *node=new ArcNode[e];for(i=0;i<e;i++)scanf("%d%d%d",&node[i].v1,&node[i].v2,&node[i].cost);sort(node,node+e,cmp);for(i=0;i<v;i++){scanf("%d",&add[i]);father[i]=i;}sort(add,add+v);Kruskal(node);//运用克鲁斯卡尔算法求出cost值最小的连通图printf("%d\n",s+add[0]);delete[] node;node=NULL;}return 0;}