找到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());
}
写不明白,最优解以后再补