字节笔试-N人相邻身高差最小

字节笔试-N人相邻身高差最小

题目

描述
n个人围成一圈，为了避免相邻两人身高差过大，需要调整每个人的位置。求最大身高差的最小值
输入
第一行为n（3<=n<=100)代表人数
接下来每有n个正整数 数的区间为[1,300]代表身高（吐槽：哪来的3米高的）
输出
最大身高差最小值

示例1
输入

4
100 150 200 250

输出

100

说明
例如150 100 200 250，此时任意两人之间的差距不大于100

题解

正确性自行验证

#include <bits/stdc++.h>

using namespace std;

int curr_max = 0;
int threshold = 1000;
int get_shold_ahead(int *arr,int idx,int len) //获取当前位置和前一个人的差值的绝对值
{
    if(idx==0)
    {
        return abs(arr[0]-arr[len-1]);
    }
    else
    {
        return abs(arr[idx]-arr[idx-1]);
    }
}

int check_ahead(int curr,int thre) //剪枝的关键，如果本次交换不能减小身高差，那就没必要继续搜索下去了
{
    return curr>=thre;
}

void dfs(int *arr,int start,int end,int (*break_search)(int,int))
{
    if(start>=end) //递归终点
    {

        /*for(int idx=0;idx<end;idx++) //这里是给调试检查结果的
            cout<<arr[idx]<<" ";
        printf("[%d]\n",curr_max);*/

        threshold = curr_max;
        return;
    }

    for(int idx=start;idx<end;idx++)
    {
        swap(arr[idx],arr[start]); //更改当前位置
        if(start==0)
        {
            curr_max = get_shold_ahead(arr,start,end);
        }
        else
        {
            curr_max = max(get_shold_ahead(arr,start,end),curr_max);
        }

        if(!break_search(curr_max,threshold)) //决定继续递归下去
        {
            dfs(arr,start+1,end,break_search);
        }
        swap(arr[idx],arr[start]);
    }
}

int main()
{
    int n = 4;
    cin >> n;
    vector<int> friends(n);
    for(int f_idx=0;f_idx<n;f_idx++)
    {
        cin >> friends[f_idx];
        //friends[f_idx] = rand() % 300 + 1;
    }
    dfs(&friends[0],0,friends.size(),check_ahead);
    cout << threshold <<endl;
    return 0;
}

这道题其实并不难，不知道当时为啥就一点思路都没有
用了函数指针，主要是个人为了复习指针相关
容器那个是试验一下能不能当成数组用（前几天才刚学的C++）