小区的路

问题 H:
小区的路
时间限制:
1Sec  内存限制:
128MB
提交: 21  解决:
6

[提交][状态][讨论版]
题目描述
你作为一名精通算法的程序员，被某公司聘请到开发区协助建设小区。小区已经建设N（1<N<100，编号1-N）幢大楼，计划任意两个大楼都能直接或间接到达，且已经对修建某些大楼之间的路做了很多预算，施工前，开发商又打算建造一座公园，而且计划每个大楼的人都能直接或者间接通过大路走到公园，同样做好了所有预算，现在让你来计算实现需求至少需要花费多少资金。
输入
第一行输入T（0<T<=100），表示测试数据的数量,接下来是两个数N，M，分别表示大楼的个数，做过预算的路的条数，接下来有N个数，第i
个数字表示第i
座大楼直接连接到公园需要的资金，接下来M行，每行包括三个数字a,b,c，表示大楼a
和b 修路，需要花费资金为c。
输出
对每组数据测试数据，输出满足大楼和公园能相互到达需要的最少资金。
样例输入

2
3 2
10 5 8
1 2 5
1 3 6
3 1
10 5 8
1 2 6

样例输出

16
19
#include <iostream>
#include <cstdio>
#include <algorithm>
#define MAXN 1000
using namespace std;
struct node{
int v;
int u;
int cost;
int st;
}g[MAXN];
int per[MAXN];
bool cmp(node g1, node g2)
{
return g1.cost < g2.cost;
}
int find(int x)
{
if(x == per[x]) return x;
else return per[x] = find(per[x]);
}
int main()
{
int n, m, t, c;
scanf("%d", &t);
while(t--)
{
scanf("%d%d", &n, &m);
int val = 0;
int w = n + m;
for (int i = 0; i <= n; i++) {
per[i] = i;
}
for (int i = 1; i <= n; i++) {
g[i + m].u = 0;
g[i + m].v = i;
scanf("%d", &c);
g[i + m].cost = c;
}
for(int i = 1; i <= m; i++)
{
scanf("%d%d%d", &g[i].u, &g[i].v, &g[i].cost);
}
sort(g + 1, g + 1 + w, cmp);
for(int i = 1 ;i <= w; i++){
int fx = find(g[i].u);
int fy = find(g[i].v);
if(fx != fy) {
val += g[i].cost;
per[fy] = fx;
}
}
printf("%d\n",val);
}
return 0;
}