洛谷 P5143 攀爬者

原题链接

题目背景

HKE考完GDOI之后跟他的神犇小伙伴们一起去爬山。

题目描述

他在地形图上标记了N个点，每个点Pi都有一个坐标(x_i,y_i,z_i)。所有点对中，高度值z不会相等。HKE准备从最低的点爬到最高的点，他的攀爬满足以下条件：

(1) 经过他标记的每一个点；

(2) 从第二个点开始，他经过的每一个点高度z都比上一个点高；

(3) HKE会飞，他从一个点Pi爬到Pj的距离为两个点的欧几里得距离。即，sqrt{(X_i-X_ j)^2+(Y_i-Y_j)^2+(Z_i-Z_j)^2}

现在，HKE希望你能求出他攀爬的总距离。

输入格式

第一行，一个整数NN表示地图上的点数。

接下来N行，三个整数x_i,y_i,z_i表示第i个点的坐标。

输出格式

一个实数，表示HKE需要攀爬的总距离（保留三位小数）

输入输出样例

输入

5
2 2 2
1 1 1
4 4 4
3 3 3
5 5 5

输出

6.928

说明/提示

对于100%的数据，1≤N≤50000，答案的范围在double范围内。

---

已AC代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1e7+5;    //数组不能开太小
struct check{
	long long int x,y,z;
}a[maxn];

bool cmp(check x,check y)
{
	return x.z<y.z;    //结构体字典序排序
}
//（若无这个函数系统会报错）

int main()
{
	double distance;
	int n,i,j;
	scanf("%d",&n);
	for(i=0;i<n;++i)
	{
		cin>>a[i].x>>a[i].y>>a[i].z;
	}
	
	sort(a,a+n,cmp);    //将输入的三个点进行排序
	
	for(i=1;i<n;++i)
	{
		distance=distance+sqrt((a[i-1].x-a[i].x)*(a[i-1].x-a[i].x)+(a[i-1].y-a[i].y)*(a[i-1].y-a[i].y)+(a[i-1].z-a[i].z)*(a[i-1].z-a[i].z));    //计算距离
	}
	
	printf("%.3lf",distance);
	return 0;
}