给定一个排序好的数组,两个整数 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来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-k-closest-elements
思路:
找到最靠近 x 的k个数,感觉不太难,但是实现起来还真的不太简单,于是我们可以转变一下思路,去除最不靠近x的len-k个数。似乎这个比刚刚那个方案好实现,而且得到的是一个连续的子序列,不需要进行排序操作。
使用两个指针,一个表示数组开始,可以表示数组结束,判断距离然后移动两个指针。最终把两个指针区域的数字返回就行了
public List<Integer> findClosestElements(int[] arr, int k, int x) {
List<Integer> res = new ArrayList<>();
if(k<=0){ // 特殊值处理
return res;
}
// len 表示arr的长度
// l 表示数组低位
// h 表示数组高位
int len = arr.length;
int l=0,h=len-1;
while(len>k){
// 判断数组地位距离x近还是高位距离x近
if(Math.abs(arr[l]-x) > Math.abs(arr[h]-x)){
l++;
}else {
h--;
}
len--;
}
// 制作结果
for(int i=l;i<=h;i++) {
res.add(arr[i]);
}
return res;
}