找到K个最接近的元素
给定一个 排序好 的数组 arr
,两个整数 k
和 x
,从数组中找到最靠近 x
(两数之差最小)的 k
个数。返回的结果必须要是按升序排好的。
整数 a
比整数 b
更接近 x
需要满足:
|a - x| < |b - x|
或者|a - x| == |b - x|
且a < b
public List<Integer> findClosestElements(int[] arr, int k, int x) {if(arr.length == 1){return new ArrayList<>(Arrays.asList(arr[0]));}int end = arr.length-1;int start = 0;if(arr[start] >= x){return Arrays.stream(Arrays.copyOfRange(arr, 0, k)).boxed().collect(Collectors.toList());}if(arr[end] <= x){return Arrays.stream(Arrays.copyOfRange(arr, arr.length-k,arr.length)).boxed().collect(Collectors.toList());}while(true){boolean flag = true;if(arr[start+1] <= x){start++;flag = false;}if(arr[end-1] >= x){end--;flag = false;}if(flag){break;}}return getClose(start,end,x,k,arr);}private List<Integer> getClose(int start, int end, int x, int k, int[] arr) {ArrayList<Integer> res = new ArrayList<Integer>(k);while(end - start -1 != k){if(end == arr.length || (start != -1 && arr[end] - x >= x - arr[start])){start--;}else{end++;}}return Arrays.stream(Arrays.copyOfRange(arr, start + 1, end)).boxed().collect(Collectors.toList());}
写不明白,最优解以后再补