小猴编程周赛C++ | 缆车建设

学习C++从娃娃抓起！记录下在学而思小猴编程学习过程中的题目，记录每一个瞬间。侵权即删，谢谢支持！

附上汇总贴：小猴编程C++ | 汇总-CSDN博客

---

【题目描述】

小猴正在开发的旅游区有 n 座山，第 i 座山的坐标为 (xi,yi)，高度为 hi，现在要在各个山顶间建立一条完整的缆车线路（即选择一个 1∼n 的排列 a1,...,an，然后在第 ai−1 座山与第 ai(2≤i≤n) 座山的山顶间建立缆线）。

小猴请你进行线路规划，请问缆车线路的总长度最小为多少？

【输入】

第一行，包含一个正整数 n。 接下来 n 行，每行三个正整数 xi,yi,hi。

【输出】

一个浮点数表示结果，四舍五入保留小数点后 2 位。

【输入样例】

3
0 0 2
1 3 5
5 5 1

【输出样例】

10.36

【代码详解】

#include <bits/stdc++.h>
using namespace std;
int n, x[15], y[15], h[15], vis[15];
double ans = 1e9;
double sq(double x)
{
    return x*x;
}
double gdis(int a, int b)
{
    if (a==0) return 0;
    return sqrt(sq(x[a]-x[b])+sq(y[a]-y[b]) + sq(h[a]-h[b]));
}
//枚举第step个山，当前路线总和是sum，上一个山的编号是last
void dfs(int step, double sum, int last)
{
    if (sum>=ans) return;  //最小值优化
    if (step>n) {
        ans = min(ans, sum);
        return ;
    }
    for (int i=1; i<=n; i++) {
        if (vis[i]==0) {
            vis[i] = 1;
            dfs(step+1, sum+gdis(last, i), i);
            vis[i] = 0;
        }
    }
}
int main()
{
    cin >> n;
    for (int i=1; i<=n; i++) {
        cin >> x[i] >> y[i] >> h[i];
    }
    dfs(1, 0, 0);
    printf("%.2lf
", ans);
    return 0;
}

【运行结果】

3
0 0 2
1 3 5
5 5 1
10.36