布线问题
时间限制: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
#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<string>
#include<vector>
#include<deque>
#include<queue>
#include<algorithm>
#include<set>
#include<map>
#include<stack>
#include<time.h>
#include<math.h>
#include<list>
#include<cstring>
#include<fstream>
//#include<memory.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define INF 1000000007
#define pll pair<ll,ll>
#define pid pair<int,double>
const int MAXV=505; //顶点个数
struct E{
int to,cost;
};
bool operator>(const E&a,const E&b){
return a.cost>b.cost;
}
vector<E>vec[MAXV]; //保存边 邻接矩阵
bool used[MAXV]; //标记是否加入了生成树
int prim(int n){
fill(used,used+n+1,false);
int res=0;
priority_queue<E,vector<E>,greater<E> >que;
used[1]=true;
for(int i=0;i<vec[1].size();++i)
que.push(vec[1][i]);
while(!que.empty()){
E f=que.top();
que.pop();
if(used[f.to]==false){
res+=f.cost;
used[f.to]=true;
for(int i=0;i<vec[f.to].size();++i){
E&t=vec[f.to][i];
if(used[t.to]==false)
que.push(t);
}
}
}
return res;
}
int main()
{
//freopen("/home/lu/文档/r.txt","r",stdin);
//freopen("/home/lu/文档/w.txt","w",stdout);
int n,V,E;
scanf("%d",&n);
while(n--){
scanf("%d%d",&V,&E);
for(int i=1;i<=V;++i)
vec[i].clear();
for(int i=0;i<E;++i){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
vec[a].push_back({b,c});
vec[b].push_back({a,c});
}
int k=INF;
for(int i=1;i<=V;++i){
int c;
scanf("%d",&c);
k=min(k,c);
}
printf("%d\n",prim(V)+k);
}
return 0;
}