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]
说明:
- k 的值为正数,且总是小于给定排序数组的长度。
- 数组不为空,且长度不超过 104
- 数组里的每个元素与 x 的绝对值不超过 104
题解
先利用二分查找的思想找到里目标值最近的数的下标
然后双指针从该下标进行两侧扩展,直至扩展到找到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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。