南阳理工OJ 题目38 布线问题

布线问题
时间限制：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;
}