(双指针、二分Binary Search) leetcode 658. Find K closest Elements
題意:給定一個升序排列的數(shù)組,找到k個與x最相近的元素(即差值最小),返回的結(jié)果必須要是按升序排好的。如果有兩個數(shù)與?x的差值一樣,優(yōu)先選擇數(shù)值較小的那個數(shù)。
?
解法一:雙指針(排除法),一個一個刪,因為是有序數(shù)組,且返回的是連續(xù)升序子數(shù)組,所以每一次刪除的元素一定是位于邊界;如果數(shù)組含有共 7 個元素,要保留 3 個元素,因此要刪除 4 個元素(arr.size()-k);因為要刪除的元素都位于邊界,于是可以使用雙指針(左指針指向數(shù)組的第一個元素,右指針指向數(shù)組最后一個元素)對撞的方式確定保留區(qū)間。
將需要刪除的元素個數(shù)作為循環(huán)的條件,當(dāng)right指向的元素減去x 大于或等于 left指向的元素減去x 的距離時,說明left指向的元素更接近x,故將right-1。(因為如果有兩個數(shù)與?x的差值一樣,優(yōu)先選擇數(shù)值較小的那個數(shù))所以等于也是將right-1; 否則 left+1。
class Solution { public:vector<int> findClosestElements(vector<int>& arr, int k, int x) {int size = arr.size(), n = size-k;int left = 0, right = size-1;while(n){if(x - arr[left] <= arr[right] - x)right--;elseleft++;n--;}vector<int> a; // for(int i=left; i<=right; ++i) // a.push_back(arr[i]);a.assign(arr.begin()+left, arr.begin()+right+1);return a;} };解法二: Binary Search
?
這道題的Binary Search 真的不好想 :( ? 下面是我整理的解題思路,參考了很多大神的思路(鏈接如下),突然發(fā)現(xiàn)二分搜索有好多變種啊,搞得我有點暈暈的。
?我用的兩個二分搜索模板:https://www.acwing.com/blog/content/31/
對應(yīng)的講解視頻:https://www.bilibili.com/video/av41422769
二分搜索github的總結(jié)貼:https://github.com/yuzhoujr/leetcode/issues/8
二分搜索的總結(jié)博客:http://blackblog.tech/2018/10/08/LeetCode2Divide/#more
youtube上講解這道題用二分搜索的視頻:https://www.youtube.com/watch?v=3ifFNvdfjyg
?
?
?
class Solution { public:vector<int> findClosestElements(vector<int>& arr, int k, int x) {int left = 0, right = arr.size()-k;while(left< right){int mid = left + (right-left)/2;if(x> arr[mid]){if(x-arr[mid] > arr[mid+k]-x)left = mid+1;elseright = mid;}elseright = mid;}return vector<int>(arr.begin()+left, arr.begin()+left+k);} };?
轉(zhuǎn)載于:https://www.cnblogs.com/Bella2017/p/11223088.html
總結(jié)
以上是生活随笔為你收集整理的(双指针、二分Binary Search) leetcode 658. Find K closest Elements的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: P1772 [ZJOI2006]物流运输
- 下一篇: MMIX机器简要介绍