布线问题
时间限制:
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<stdio.h> #include<string.h> #include<math.h> #include<algorithm> using namespace std; #define INF 0x3f3f3f3f #define MIN(a,b) a>b?b:a int n,m; int map[510][510],dis[510],vis[510],s[510]; void prim(int x) { int i,j,k,min; int sum=0; memset(vis,0,sizeof(vis)); for(i=1;i<=n;i++) dis[i]=map[1][i]; dis[x]=0; vis[x]=1; for(j=1;j<n;j++) { k=1; min=INF; for(i=1;i<=n;i++) { if(!vis[i]&&dis[i]<min) { min=dis[i]; k=i; } } vis[k]=1; sum+=min; for(i=1;i<=n;i++) { if(!vis[i]&&dis[i]>map[k][i]) dis[i]=map[k][i]; } } printf("%d\n",sum+s[1]); } int main() { int t; int i,j,k; int a,b,c; scanf("%d",&t); while(t--) { scanf("%d%d",&n,&m); memset(map,INF,sizeof(map)); while(m--) { scanf("%d%d%d",&a,&b,&c); if(map[a][b]>c) map[a][b]=map[b][a]=c; } int mm=INF; for(i=1;i<=n;i++) scanf("%d",&s[i]); sort(s+1,s+n+1); prim(1); } return 0; }
-
第一行是一个整数n表示有n组测试数据。(n<5)