洛谷P5143 攀爬者

题目背景

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

题目描述

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

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

(2) 从第二个点开始,他经过的每一个点高度 zz 都比上一个点高;

(3) HKE 会飞,他从一个点 PiP_i 爬到 PjP_j 的距离为两个点的欧几里得距离。即,(XiXj)2+(YiYj)2+(ZiZj)2\sqrt{(X_i-X_j)^2+(Y_i-Y_j)^2+(Z_i-Z_j)^2}

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

输入格式

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

接下来 NN 行,三个整数 xi,yi,zix_i,y_i,z_i 表示第 ii 个点的坐标。

输出格式

一个实数,表示 HKE 需要攀爬的总距离(保留三位小数)

样例 #1

样例输入 #1

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

样例输出 #1

6.928

提示

对于100%的数据,1N500001\leq N\leq 50000,答案的范围在 double 范围内。

题解

#include <bits/stdc++.h>
using namespace std;

int n;
double ans;

struct locate {
	int x, y, z;
} l[50010];

bool cmp(locate x, locate y) {
	return x.z < y.z;
}

int main() {
	cin >> n;
	for (int i = 0; i < n; i++)
		cin >> l[i].x >> l[i].y >> l[i].z;
	sort(l, l + n, cmp);
	for (int i = 1; i < n; i++)
		ans += sqrt((l[i].x - l[i - 1].x) * (l[i].x - l[i - 1].x) + (l[i].y - l[i - 1].y) * (l[i].y - l[i - 1].y) +
		            (l[i].z - l[i - 1].z) * (l[i].z - l[i - 1].z));
	printf("%.3lf", ans);
	return 0;
}