每日一题---村庄建设

问题描述
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。
题解
很明显，这是一道最小生成树的模版题。1-n任意两个点之间的边为题目的给出的式子，然后可以建邻接矩阵用Prim写，也可以用结构体存边用Kruskal写。(具体模版代码可以百度查到)
我这里的代码直接用的是Kruskal。。

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
const int maxn=1e6+5;
using namespace std;
struct edge{
	int u;
	int v;
	double w;
}e[maxn];
bool cmp(edge a,edge b){
	return a.w<b.w;
}              	
int father[1005];
double x[1005],y[1005],z[1005];
int find(int x){
	if(father[x]==x){
		return x;
	}
	return father[x]=find(father[x]);
}
double dis(int i,int j){
	return sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]))+(z[i]-z[j])*(z[i]-z[j]);
}
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%lf%lf%lf",&x[i],&y[i],&z[i]);
		father[i]=i;
	}
	int cnt=0;
	for(int i=1;i<=n;i++){
		for(int j=i+1;j<=n;j++){
			e[cnt].u=i;
			e[cnt].v=j;
			e[cnt].w=dis(i,j);
			cnt++;
		}
	}
	sort(e,e+cnt,cmp);
	double ans=0.0;
	for(int i=0;i<cnt;i++){
		int p=find(e[i].u);
		int q=find(e[i].v);
		if(p==q){
			continue;
		}
		ans+=e[i].w;
		father[q]=p;
	}
	printf("%.2f",ans);
	return 0;
}

*tips:为什么这里边的数组要开1e6+5这么大？这里n最大取1000，因为要考虑最大边数，而在无向图为完全图时，取得的边数最多，此时任意两个顶点间都有直接边，边的条数为：（n-1）+…+2+1=n(n-1)/2，即e=499500，所以这里直接开百万容量的数组比较合适。