布线问题
时间限制:
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开始),由于安全问题,只能选择一个楼连接到外界供电设备。
数据保证至少存在一种方案满足要求。
输出
- 每组测试数据输出一个正整数,表示铺设满足校长要求的线路的最小花费。 样例输入
-
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
-
-
#include <iostream> #include <cstdio> #include <string.h> #include <algorithm> using namespace std; const int N=500+10; const int M=N*(N-1)/2 ; struct Node{ int s,end,w; }node[M]; bool cmp(Node a,Node b){ return a.w<b.w; } int pre[N],n,m; int find(int x){ int r=x; while(pre[r]!=r) r=pre[r]; int i=x,j; while(i!=r){ j=pre[i]; pre[i]=r; i=j; } return r; } inline void Init(int n){ for(int i=1;i<=n;i++) pre[i]=i; } inline bool join(int x,int y){ int fx=find(x); int fy=find(y); if(fx!=fy){ //把x和y加入统一个集合 pre[fx]=fy; return true; } return false; } int kruskal() { int i,count1=1,sum=0; Init(n); //并查集初始化,每个节点到父亲是自己 //int x,y; sort(node,node+m,cmp); //快速排序 for(i=0;count1<n;i++) { //最小生成树 n个点 就n-1条边 if(join(node[i].s,node[i].end)){ sum+=node[i].w; count1++; //加上一条边 } } // x=find(edge[i].s); //点edge[i].u的父亲是x // y=find(edge[i].end); //点edge[i].v的父亲是y /* if(x!=y) //判断是否会路 { count++; //加上一条边 p[x]=y; //把x和y加入统一个集合 sum+=edge[i].w; }*/ return sum; } int main(){ int t,i,j,small; scanf("%d",&t); while(t--){ small=99999999; scanf("%d%d",&n,&m); for(i=0;i<m;i++) scanf("%d%d%d",&node[i].s,&node[i].end,&node[i].w); for(i=0;i<n;i++){ scanf("%d",&j); if(small>j) small=j; } printf("%d\n",kruskal()+small); } return 0; }
-
第一行是一个整数n表示有n组测试数据。(n<5)