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);} };

總結

以上是生活随笔為你收集整理的LeetCode 658. 找到 K 个最接近的元素（二分查找）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。