力扣：658. 找到 K 个最接近的元素

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第27天，点击查看活动详情

658. 找到 K 个最接近的元素

给定一个 排序好 的数组 arr ，两个整数 k 和 x ，从数组中找到最靠近 x（两数之差最小）的 k 个数。返回的结果必须要是按升序排好的。

整数 a 比整数 b 更接近 x 需要满足：

• |a - x| < |b - x| 或者
• |a - x| == |b - x| 且 a < b

示例 1：
输入：arr = [1,2,3,4,5], k = 4, x = 3
输出：[1,2,3,4]

示例 2：
输入：arr = [1,2,3,4,5], k = 4, x = -1
输出：[1,2,3,4]
 

 提示：

• 1 <= k <= arr.length
• 1 <= arr.length <= 104
• arr 按 升序 排列
• -104 <= arr[i], x <= 104

问题解析

自定义排序

我们可以想办法，把最接近x的数都挪放到数组的前面来，这样最前面的k个数就是我们要的答案了。

可以用自定义排序规则cmp来实现这一过程，对于数组arr中的两个数a,b来说：

• 如果abs(x-a)<abs(x-b),那我们可以把数字a放到b的前面，因为a显然比b更接近x。
• 如果abs(x-a)==abs(x-b),那我们把a和b中两者较小的部分放在前面。

（其实可以看出，这其实就是题目给的两个条件） 依照自定义排序对arr数组排序后，我们直接取出前面的k个元素即可。 要注意的是，题目要返回的是升序序列，而自定义排序可能会把顺序搞混，所以我们还要对结果排序后再返回。

AC代码

int num;
bool cmp(int a,int b)
{
    int x=abs(a-num),y=abs(b-num);
    if(x!=y)return x<y;
    return a<b;
}
class Solution {
public:
    
    
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        num=x;

        sort(arr.begin(),arr.end(),cmp);
        vector<int>res;
        for(int i=0;i<k;i++)res.push_back(arr[i]);

        return res;
    }
};