【通电】（并查集+最小生成树）

2015年，全中国实现了户户通电。作为一名电力建设者，小明正在帮助一带一路上的国家通电。
  这一次，小明要帮助 n 个村庄通电，其中 1 号村庄正好可以建立一个发电站，所发的电足够所有村庄使用。
  现在，这 n 个村庄之间都没有电线相连，小明主要要做的是架设电线连接这些村庄，使得所有村庄都直接或间接的与发电站相通。
  小明测量了所有村庄的位置（坐标）和高度，如果要连接两个村庄，小明需要花费两个村庄之间的坐标距离加上高度差的平方，形式化描述为坐标为 (x_1, y_1) 高度为 h_1 的村庄与坐标为 (x_2, y_2) 高度为 h_2 的村庄之间连接的费用为：
sqrt((x_1-x_2)(x_1-x_2)+(y_1-y_2)(y_1-y_2))+(h_1-h_2)*(h_1-h_2)。
  在上式中 sqrt 表示取括号内的平方根。请注意括号的位置，高度的计算方式与横纵坐标的计算方式不同。
  由于经费有限，请帮助小明计算他至少要花费多少费用才能使这 n 个村庄都通电。
【输入格式】
输入的第一行包含一个整数 n ，表示村庄的数量。
接下来 n 行，每个三个整数 x, y, h，分别表示一个村庄的横、纵坐标和高度，其中第一个村庄可以建立发电站。
【输出格式】
输出一行，包含一个实数，四舍五入保留 2 位小数，表示答案。
【样例输入】
4
1 1 3
9 9 7
8 8 6
4 5 4
【样例输出】
17.41
【评测用例规模与约定】
对于 30% 的评测用例，1 <= n <= 10；
对于 60% 的评测用例，1 <= n <= 100；
对于所有评测用例，1 <= n <= 1000，0 <= x, y, h <= 10000。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <set>
#include <cmath>
#include <vector>
using namespace std;
int n;
int fa[1005], rankk[1005];
struct node {
	int x, y, h;
}no[1005];
double consume(node a,node b) {
	return sqrt((a.x - b.x) * (a.x - b.x) + (a.y - b.y) * (a.y - b.y)) + (a.h - b.h) * (a.h - b.h);
}
struct edge {
	int u, v;
	double w;
}edg[1000000];
int find(int x) {
	if (x == fa[x])return x;
	else return fa[x] = find(fa[x]);
}
bool cmp(edge a,edge b) {
	return a.w < b.w;
}
int main()
{
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> no[i].x >> no[i].y >> no[i].h;
	}
	for (int i = 1; i <= n; i++) {
		fa[i] = i;
	}
	int index = 1;
	for (int i = 1; i <= n; i++) {
		for (int j = i + 1; j <= n; j++) {
			edg[index].u = i;
			edg[index].v = j;
			edg[index].w=consume(no[i], no[j]);
			index++;
		}
	}
	sort(edg + 1, edg + index, cmp);
	double sum = 0;
	for (int i = 1; i <= index - 1; i++) {
		int x = find(edg[i].u), y = find(edg[i].v);
		if (x != y) {
			fa[x] = y;
			sum += edg[i].w;
		}
	}
	printf("%.2lf", sum);
}

以并查集模板题和最小生成树模板题为基础，熟练之后可以直接一气呵成！！！！！！！！！无他，惟手熟尔，仅此而已。