LeetCode 658. 找到 K 个最接近的元素（C++）

给定一个排序好的数组，两个整数 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 的值为正数，且总是小于给定排序数组的长度。

数组不为空，且长度不超过 10^4

数组里的每个元素与 x 的绝对值不超过 10^4

C++

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) 
    {
        int n=arr.size();
        int left=0;
        int right=n-1;
        int idx=0;
        int flag=0;
        while(left<right)
        {
            int mid=(left+right)/2;
            if(arr[mid]==x)
            {
                flag=1;
                idx=mid;
                break;
            }
            else if(arr[mid]>x)
            {
                right=mid-1;
            }
            else
            {
                left=mid+1;
            }
        }
        if(!flag)
        {
            idx=left;
        }
        int low=max(0,idx-k+1);
        int high=min(n-1,idx+k-1);
        while(high-low>k-1)
        {
            if(abs(arr[low]-x) > abs(arr[high]-x))
            {
                low++;
            }
            else
            {
                high--;
            }
        }
        return vector<int>(arr.begin()+low,arr.begin()+high+1);
    }
};