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

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]

说明:

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

题解

先利用二分查找的思想找到里目标值最近的数的下标
然后双指针从该下标进行两侧扩展，直至扩展到找到k个数。详细过程见代码

代码

	int findK(vector<int>& nums,int k){			//找到离k最近的数的下标
        int l=0,r=nums.size()-1,mid;
        while(l <= r){
            mid = (l+r)/2;
            if(nums[mid] == k)  return mid;
            else if(nums[mid]<k)  l = mid+1;
            else r = mid-1;
        }
        if(l == 0)  return 0;
        else if(l == nums.size())   return l-1;
        return k-nums[l-1]>nums[l]-k? l : l-1;
    }
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        int l = findK(arr,x);	
        int r = l;
        k--;
        while(k){
            if(l==0){
                r += k;
                break;
            }else if(r == arr.size()-1){
                l -= k;
                break;
            }
            if(x-arr[l-1] <= arr[r+1]-x){
                l--;
            }else{
                r++;
            }
            k--;
        }
        return vector<int>(arr.begin()+l,arr.begin()+r+1);
    }

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