布線問題

時間限制：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

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 <cstring>  
#include <cmath>  
#include <algorithm>  
using namespace std ;  
  
const int MAXN = 505 ;  
const int INF = 0xfffffff ;  
int mat[MAXN][MAXN] ;  
int vset[MAXN] ;  
int add[MAXN] ;  
int v, e, ans ;  
  
struct Node  
{  
    int a, b ;  
    int w ;  
};  
Node MST[MAXN*MAXN] ;  
  
bool cmp (const Node &t1, const Node &t2)  
{  
    return t1.w < t2.w ;  
}  
  
void init ()  
{  
    int i, k = 1 ;  
    int c, d, t ;  
    scanf ("%d%d", &v, &e) ;  
  
    for (i = 1; i <= e; i ++)  
    {  
        scanf ("%d%d%d", &c, &d, &t) ;  
        MST[k].a = c ;  
        MST[k].b = d ;  
        MST[k].w = t ;  
        k++ ;  
    }  
    for (i = 1; i <= v; i ++)  
    {  
        vset[i] = i ;  
        scanf ("%d", &add[i]) ;  
    }  
  
    sort (add+1, add+v+1) ;  
    sort (MST+1, MST+1+e, cmp) ;  
}  
  
void kruscal ()  
{  
    int i, j, tag = 1 ;  
    int num = 1 ;  
    int u1, v1, sn1, sn2 ;  
    ans = 0 ;  
  
    while (num < v)  
    {  
        u1 = MST[tag].a ;  
        v1 = MST[tag].b ;  
        sn1 = vset[u1] ;  
        sn2 = vset[v1] ;  
  
        if (sn1 != sn2)  
        {  
            num ++ ;  
            ans += MST[tag].w ;  
            for (i = 1; i <= v; i ++)  
                if (vset[i] == sn2)  
                    vset[i] = sn1 ;  
  
        }  
        tag ++ ;  
    }  
  
}  
  
int main ()  
{  
    int tcase ;  
    scanf ("%d", &tcase) ;  
    while (tcase --)  
    {  
        init () ;  
        kruscal () ;  
        printf ("%d\n", ans + add[1]) ;  
    }  
    return 0 ;  
}          

nyoj 38布線問題