[蓝桥杯][第十一届蓝桥杯省模拟赛]通电（Java代码）

【问题描述】
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。
【代码】

package Hello11;

/*
 * 以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)为权值
 * 用Prim算法得到最小生成树即为答案。
 */

import java.util.*;
public class Main9 {
	public static double dist[][] = new double[1005][1005]; //两点之间的距离
	public static int vist[] = new int[1005]; //点是否已经加入树
	public static double lowc[] = new double[1005]; //点离树的距离
	public static Node node[] = new Node[1005]; //保存输入数据
	public static class Node{
		int x;
		int y;
		int h;
		public Node(int x,int y,int h) {
			this.x = x;
			this.y = y;
			this.h = h;
		}
	}
	public static double Prim(int n) {
		double price = 0; //最小花费
		for(int i=1;i<=n;i++) { //以村庄1为起点，lowc[i]初始化为每个村庄i到村庄1的距离
			lowc[i]=dist[1][i];
		}
		for(int i=1;i<=n;i++) { //n个点:所以n次循环
			double min = 1000000000; //初始化最小值为无穷大
			int p = -1; //p记录本次循环距离最小生成树最近的点
			for(int j=1;j<=n;j++) {
				if(vist[j]==0 && lowc[j]<min) { //如果没有加入树，并且到树的距离比min小，
					min = lowc[j]; //则更新min
					p = j; //更新p
				}
			}
			if(min==1000000000) return -1; //如果min仍为无穷大，则该图为不连通图
			vist[p] = 1; //标记村庄p加入最小生成树
			price+=min; //更新花费
			for(int j=1;j<=n;j++) { //如果未加入的村庄j，通过加入的村庄p距离更近了
				if(vist[j]==0 && lowc[j]>dist[p][j]) {
					lowc[j] = dist[p][j]; //则更新lowc[j]
				}
			}
		}
		return price;
	}
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		for(int i=1;i<=n;i++) {
			node[i] = new Node(sc.nextInt(),sc.nextInt(),sc.nextInt());
		}
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=n;j++) {		
				double len = Math.sqrt(
						 (node[i].x-node[j].x)*(node[i].x-node[j].x)
						+(node[i].y-node[j].y)*(node[i].y-node[j].y))
						+(node[i].h-node[j].h)*(node[i].h-node[j].h);
				dist[i][j]=len;
			}
		}
		double result = (double)((int)(Prim(n)*100))/100; //结果保留两位小数
		//String result = String.format("%.2f",Prim(n));
		System.out.print(result);
	}
}

/*
    java保留两位小数
	double data = 3.02;
    String result = String.format("%.1f",data);
    System.out.println(result);//输出3.0
*/