蓝桥杯模拟赛第十题

问题描述
　　受大西线调水工程启发，小明也准备设计一条调水的水渠。
　　小明经费有限，他只能在一块有限区域内建立一条简单的水渠。
　　小明首先勘探了地形，在这块地中有一处水源，必须用作水渠的起点。另外，小明还测量了一些点，包括点的位置和高度。如果两个小明测量的点之间的距离不超过 d 且高度不同，小明就可以在这两点之间建立一段水渠，让水从高处流向低处，这一段的长度为两点之间的直线距离（即将横坐标的差的平方加上纵坐标的差的平方加上高度差的平方后再开平方根）。
　　小明计划只修一条主水渠，不建立分支的水渠。由于水渠能影响的范围与水渠的长度正相关，小明希望水渠尽可能长。
　　请注意，水渠必须从水源开始修，并且高度应当递减。水渠的不同段可能交叉（建个桥即可）。
输入格式
　　输入的第一行包含一个整数 n ，表示小明已经测量的点数。
　　接下来 n 行，每行三个整数 x, y, h，分别表示测量的点坐标为 (x, y)，高度为 h。这部分的第一个点即为水源，第一个点的h值大于其他点的h值。
　　接下来一行包含一个整数 d。
输出格式
　　输出一行，包含一个实数，四舍五入保留 2 位小数，表示水渠最长能修多长。
样例输入
5
1 1 10
2 3 8
4 5 9
1 2 5
4 5 5
8
样例输出
10.66
样例说明
　　在这些点中有两个坐标为 (4, 5) 的点，这是允许的。
评测用例规模与约定
　　对于 30% 的评测用例，1 <= n <= 10；
　　对于 60% 的评测用例，1 <= n <= 20；
　　对于所有评测用例，1 <= n <= 1000，0 <= h <= 10000，0 <= x, y <= 10000，0 < d < 1e7（10的7次方）。

 

#include<bits/stdc++.h>
using namespace std;
#define maxn 1000
#define eps 1e-8
struct node{
    int x,y,h;
};
node water[maxn];
int vis[maxn];
int cnt,n,d;
double ans,max_ans;
void add_node(int x,int y,int h){
    water[cnt].x=x;water[cnt].y=y;water[cnt].h=h;
    cnt++;
}
bool cmp(node a,node b){
    return a.h>b.h;
}

void dfs(int t,int step){
    if(step>=n-1) return;
    for(int i=t+1;i<n;i++){//到往高度比t小的点
    if(water[i].h>=water[t].h) break;
        double dis=sqrt((water[t].x-water[i].x)*(water[t].x-water[i].x)+(water[t].y-water[i].y)*(water[t].y-water[i].y));
        //cout<<d<<" "<<dis<<endl;
		if(((d-dis>=eps)||(d-dis)<=eps) &&(!vis[i])){
            vis[i]=1;
            ans+=dis; 
            max_ans=max(ans,max_ans);
            dfs(i,step+1);
            vis[i]=0;
            ans-=dis;
        }
    }
}
int main(){
    int x,y,h;//都为整数
    cin>>n;
    cnt=0;
    for(int i=0;i<n;i++){
        cin>>x>>y>>h;
        add_node(x,y,h);
    }
    cin>>d;
    sort(water,water+n,cmp);//高度从高到低
    max_ans=ans=0;
    memset(vis,0,sizeof(vis));
    vis[0]=1;
    dfs(0,0);//从起点开始走
    printf("%.2lf\n",max_ans);
    return 0;
}
/*
2
1 1 9
4 5 3
5
*/