布线问题
时间限制: 1 Sec 内存限制: 64 MB
题目描述
南阳理工学院要进行用电线路改造,现在校长要求设计师设计出一种布线方式,该布线方式需要满足以下条件:
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
来源
//prim算法
/*
注意:
1、初始化时
2、简单使用prim算法
3、在村落中找出花费最小的
4、细节:不要把 if(map[flag][k]<lowcost[k]&&lowcost[k]!=0 )&&lowcost[k]!=0放到这里for(int k=2;k<=m;k++)
*/
#include<cstdio>
#include<iostream>
#include<algorithm>
#define inf 0x3f3f3f3f
#define N 1010
using namespace std;
int lowcost[N];
int visit[N];
int map[N][N];
int cun[N];
void init()
{
}
int prim(int map[N][N],int m)
{
int sum=0;
for(int i=1;i<=m;i++)
lowcost[i]=map[1][i];
lowcost[1]=0;
for(int i=2;i<=m;i++)
{
int min=inf;
int flag=1;
for(int j=2;j<=m;j++)
{
/// printf("****\n");
if(min>lowcost[j]&&lowcost[j]!=0)
{
min=lowcost[j];
flag=j;
//printf("%d***\n",j);
}
}
sum=sum+lowcost[flag];
lowcost[flag]=0;
for(int k=2;k<=m;k++)
{
if(map[flag][k]<lowcost[k]&&lowcost[k]!=0 )
{
lowcost[k]=map[flag][k];
}
}
}
return sum;
}
int main()
{
int t;
int m,n;
int u,v,w;
int i,j;
cin>>t;
while(t--)
{
cin>>m>>n;
for( i=1;i<=m;i++)
for( j=1;j<=m;j++)
map[i][j]=inf;
for(i=0;i<n;i++)
{
cin>>u>>v>>w;
if(map[u][v]>w)
map[u][v]=map[v][u]=w;
}
int k=prim(map,m);
int a;
int minn=inf;
for(int i=1;i<=m;i++)
{
scanf("%d",&a);
minn=min(minn,a);///保留与外界相连最少的费用
}
//printf("%d\n",cun[0]);
printf("%d\n",k+minn);
}
}