2020年十一届省赛蓝桥杯本科模拟赛第九题解 村庄通电

第九题：村庄通电

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。

【思路】
首先需要根据x轴的值排序，这是可以产生最短的距离；
接着根据公式循环计算两地之间的距离即可。

【注意】
注意是为((x_1-x_2)(x_1-x_2)+(y_1-y_2)(y_1-y_2))开平方，不要加上(h_1-h_2)*(h_1-h_2)一起开平方，否则结果出错。我开始写的时候就没有看清，然后一直纠结哪里错了。

【代码】

#include<iostream>
#include<cmath>
#include<cstdio>
using namespace std;
int a[1005][3];
int n;
void sort(){
 int t_x,t_y,t_h;
 for(int i=0;i<n;i++){
  for(int j=i+1;j<n;j++){
   if(a[i][0] > a[j][0]){
    t_x = a[i][0];
    t_y = a[i][1];
    t_h = a[i][2];
    a[i][0] = a[j][0];
    a[i][1] = a[j][1];
    a[i][2] = a[j][2];
    a[j][0] = t_x;
    a[j][1] = t_y;
    a[j][2] = t_h;
   }
  }
 }
 return ;
}
int main(){
 
 double sum;
 cin>>n;
 for(int i=0;i<n;i++){
  cin>>a[i][0]>>a[i][1]>>a[i][2];
 }
 sort();
 for(int i=1;i<n;i++){
  sum+=sqrt( (a[i][0]-a[i-1][0]) * (a[i][0]-a[i-1][0]) + (a[i][1]-a[i-1][1]) * (a[i][1]-a[i-1][1])) +(a[i][2]-a[i-1][2]) * (a[i][2]-a[i-1][2]) ;
 }
 printf("%.2f\n",sum);
 return 0;
}