给定一个 排序好 的数组 arr ,两个整数 k 和 x ,从数组中找到最靠近 x(两数之差最小)的 k 个数。返回的结果必须要是按升序排好的。
整数 a 比整数 b 更接近 x 需要满足:
|a - x| < |b - x| 或者
|a - x| == |b - x| 且 a < b
示例 1:
输入:arr = [1,2,3,4,5], k = 4, x = 3
输出:[1,2,3,4]
示例 2:
输入:arr = [1,2,3,4,5], k = 4, x = -1
输出:[1,2,3,4]
解题思路:
(1)数组长度为 n,数组arr 已经按照升序排序,我们可以将数组 arr 分成两部分,前一部分所有元素都小于 x,后一部分所有元素都大于等于 x,可以通过二分查找获得。
(2)由于x将左右分为两部分指向x且接近x的数组,最后通过双指针,得出数组区间 [ l , l+k ] ,最后把区间内部的数组依次添加进list里面return即可。
class Solution {
public List<Integer> findClosestElements(int[] arr, int k, int x) {
int l = 0;
int r = arr.length - k;
while(l < r){
int mid = (l + r)/2;
if(x - arr[mid] > arr[mid + k] - x){
l = mid + 1;
}else{
r = mid;
}
}
List<Integer> list = new ArrayList<>();
for(int i = l; i < l + k;i++){
list.add(arr[i]);
}
return list;
}
}