LeetCode 658. 找到 K 个最接近的元素（二分查找）

1. 题目

给定一个排序好的数组，两个整数 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

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

2. 解题

在这里插入图片描述

class Solution {
public:vector<int> findClosestElements(vector<int>& arr, int k, int x) {if(x <= arr[0])return vector<int>(arr.begin(), arr.begin()+k);if(x >= arr[arr.size()-1])return vector<int>(arr.end()-k, arr.end());int i, left, right, disLeft, disRight;vector<int> ans;for(i = 0; i < arr.size(); ++i){if(x == arr[i]){ans.push_back(arr[i]);left = i-1;right = i+1;break;}if(x < arr[i]){left = i-1;right = i;break;}}//找x左右的位置while(left >= 0 && right <= arr.size()-1 && ans.size() < k){disLeft = x - arr[left];//左边距离disRight = arr[right] - x;//右边距离if(disLeft <= disRight){ans.push_back(arr[left]);--left;}else{ans.push_back(arr[right]);++right;}}if(ans.size() != k)//一边到头了{if(left == -1){while(ans.size() != k)ans.push_back(arr[right++]);}else//right = ans.size(){while(ans.size() != k)ans.push_back(arr[left--]);}}sort(ans.begin(), ans.end());return ans;}
};

二分法

class Solution {
public:vector<int> findClosestElements(vector<int>& arr, int k, int x) {int lo = 0, hi = arr.size() - k;while (lo < hi) {int mid = lo + (hi - lo >> 1);if (abs(x - arr[mid]) > abs(arr[mid + k] - x ))lo = mid + 1;elsehi = mid;}return vector<int>(arr.begin() + lo, arr.begin() + lo + k);}
};