Leetcode 658：找到K个最接近的元素

题目描述

给定一个排序好的数组，两个整数 k 和 x，从数组中找到最靠近 x（两数之差最小）的 k 个数。返回的结果必须要是按升序排好的。如果有两个数与 x 的差值一样，优先选择数值较小的那个数。

示例 1:

输入: [1,2,3,4,5], k=4, x=3
输出: [1,2,3,4]
 

示例 2:

输入: [1,2,3,4,5], k=4, x=-1
输出: [1,2,3,4]
 

说明:

k 的值为正数，且总是小于给定排序数组的长度。
数组不为空，且长度不超过 104
数组里的每个元素与 x 的绝对值不超过 104
 

更新(2017/9/19):
这个参数 arr 已经被改变为一个整数数组（而不是整数列表）。 请重新加载代码定义以获取最新更改。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-k-closest-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        vector<int> ans;
        int l = 0,r = 0,len = arr.size();
        int idx = lower_bound(arr.begin(),arr.end(),x) - arr.begin();
        if(idx == 0){
            l = -1;
            r = k;
        }
        else if(idx == len){
            r = len;
            l = len - k - 1;
        }
        else{
            l = idx-1;
            r = idx;
            while(k){
                if(l>=0 && r<len){
                    if(x-arr[l] <= arr[r]-x) l--;
                    else r++;
                }
                else if(l>=0) l--;
                else r++;
                k--;
            }
        }
        for(int i=l+1;i<r;i++) ans.push_back(arr[i]);
        return ans;
    }
};