每日算法系列【LeetCode 658】找到 K 个最接近的元素

简介: 给定一个排序好的数组,两个整数 k 和 x,从数组中找到最靠近 x(两数之差最小)的 k 个数。返回的结果必须要是按升序排好的。如果有两个数与 x 的差值一样,优先选择数值较小的那个数。

题目描述


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

题解


滑动窗口



image.png

image.png

二分+滑动窗口



image.png

二分



image.png

代码


滑动窗口(c++)

classSolution {
public:   
vector<int>findClosestElements(vector<int>&arr, intk, intx)
    {      
intn=arr.size();   
intl=0, r=n-1;   
while (r-l>=k) {      
if (x-arr[l] <=arr[r]-x) r--;             
elsel++;    
        }       
vector<int>res(k);   
copy(arr.begin()+l, arr.begin()+l+k, res.begin());    
returnres;  
    }
};

二分+滑动窗口(c++)

classSolution {
public:  
vector<int>findClosestElements(vector<int>&arr, intk, intx) 
    {      
intn=arr.size();    
intl=0, r=n-1;     
while (l<r) {       
intm= (l+r) /2;     
if (arr[m] <x) l=m+1;     
elser=m;    
        }       
r=min(n-1, l+k-1); 
l=max(0, l-k);   
while (r-l>=k) { 
if (x-arr[l] <=arr[r]-x) r--;     
elsel++;     
        }      
vector<int>res(k);      
copy(arr.begin()+l, arr.begin()+l+k, res.begin()
            );    
returnres;    
    }
};

二分(c++)



classSolution {
public:   
vector<int>findClosestElements(vector<int>&arr, intk, intx) 
    {      
intn=arr.size();   
intl=0, r=n-k;   
while (l<r) {      
intm= (l+r) /2;    
if (x-arr[m] >arr[m+k]-x) l=m+1;   
elser=m;    
        }      
vector<int>res(k);    
copy(arr.begin()+l, arr.begin()+l+k, res.begin()
            ); 
returnres; 
    }
};

滑动窗口(python)


classSolution:  
deffindClosestElements(self, arr: List[int], k: int, x: int) ->List[int]:    
n=len(arr)   
l, r=0, n-1whiler-l>=k:    
ifx-arr[l] <=arr[r]-x:   
r-=1else:    
l+=1returnarr[l:l+k]

二分+滑动窗口(python)



classSolution: 
deffindClosestElements(self, arr: List[int], k: int, x: int) ->List[int]:    
n=len(arr)    
l, r=0, n-1whilel<r:    
m= (l+r) //2ifarr[m] <x:   
l=m+1else:    
r=mr=min(n-1, l+k-1)    
l=max(0, l-k)  
whiler-l>=k:  
ifx-arr[l] <=arr[r]-x:   
r-=1else:    
l+=1returnarr[l:l+k]

二分(python)


classSolution:    
deffindClosestElements(self, arr: List[int], k: int, x: int) ->List[int]:    
n=len(arr)  
l, r=0, n-kwhilel<r:   
m= (l+r) //2ifx-arr[m] >arr[m+k]-x:  
l=m+1else:     
r=mreturnarr[l:l+k]

image.png

作者简介:godweiyang知乎同名华东师范大学计算机系硕士在读,方向自然语言处理与深度学习喜欢与人分享技术与知识,期待与你的进一步交流~

相关文章
【力扣】-- 移除链表元素
【力扣】-- 移除链表元素
170 1
|
9月前
|
机器学习/深度学习 存储 算法
【LeetCode 热题100】347:前 K 个高频元素(详细解析)(Go语言版)
这篇文章详细解析了力扣热题 347——前 K 个高频元素的三种解法:哈希表+小顶堆、哈希表+快速排序和哈希表+桶排序。每种方法都附有清晰的思路讲解和 Go 语言代码实现。小顶堆方法时间复杂度为 O(n log k),适合处理大规模数据;快速排序方法时间复杂度为 O(n log n),适用于数据量较小的场景;桶排序方法在特定条件下能达到线性时间复杂度 O(n)。文章通过对比分析,帮助读者根据实际需求选择最优解法,并提供了完整的代码示例,是一篇非常实用的算法学习资料。
571 90
【LeetCode 27】347.前k个高频元素
【LeetCode 27】347.前k个高频元素
147 0
|
存储 算法
LeetCode第83题删除排序链表中的重复元素
文章介绍了LeetCode第83题"删除排序链表中的重复元素"的解法,使用双指针技术在原链表上原地删除重复元素,提供了一种时间和空间效率都较高的解决方案。
LeetCode第83题删除排序链表中的重复元素
【LeetCode 06】203.移除链表元素
【LeetCode 06】203.移除链表元素
135 0
【LeetCode-每日一题】移除元素
【LeetCode-每日一题】移除元素
109 0
LeetCode第16题最接近的三数之和
该文章介绍了 LeetCode 第 16 题最接近的三数之和的解法,与第 15 题类似,通过双指针法减少循环次数,根据差值的绝对值来更新最接近的和,并总结了双指针可减少循环次数的要点。
|
算法 索引
LeetCode第34题在排序数组中查找元素的第一个和最后一个位置
这篇文章介绍了LeetCode第34题"在排序数组中查找元素的第一个和最后一个位置"的解题方法,通过使用双指针法从数组两端向中间同时查找目标值,有效地找到了目标值的首次和最后一次出现的索引位置。
LeetCode第34题在排序数组中查找元素的第一个和最后一个位置
|
Python
【Leetcode刷题Python】剑指 Offer II 082. 含有重复元素集合的组合
解决LeetCode平台《剑指 Offer II 082. 含有重复元素集合的组合》题目的Python代码实现,通过深度优先搜索算法找出所有和为特定目标值的数字组合,并在搜索过程中通过排序和跳过重复元素来避免解集中出现重复组合。
141 2
|
Python
【Leetcode刷题Python】16. 最接近的三数之和
解决LeetCode "最接近的三数之和" 问题的Python实现,通过排序和双指针法,记录并更新与目标值最接近的三数之和。
141 1

热门文章

最新文章