找到 K 个最接近的元素(java,算法)
给定一个 排序好 的数组 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; } }