* 排序 * 按照离x距离的远近进行排序 * 排序之后,就会按照距离近的在前面,远得在后面,然后直接截取就行 * 截取之后在排序,就是我们的结果啦 * 排序的时间复杂度是logn,遍历添数组是n总共是nlogn * 空间复杂度是logn
public List<Integer> findClosestElements(int[] arr, int k, int x) {
List<Integer> res = new ArrayList<>();
for (int i : arr) {
res.add(i);
}
Collections.sort(res, (a, b) ->{
if (Math.abs(a - x) != Math.abs(b - x) ){
return Math.abs(a - x) - Math.abs(b - x);
}
return a - b;
});
List<Integer> list = res.subList(0, k);
Collections.sort(list);
return list;
}
* 二分解法,双指针 * 1.首先找到最接近x的两个点,分别定义为left和right * 2.注意,当left小于0 那个只需要加right就行,当right大于数组的长度的时候,只需要减left * 3.判断那边距离更近,添加哪边
public List<Integer> findClosestElements2(int[] arr, int k, int x) {
int right = binary(arr,x);
int left = right - 1;
while (k-- > 0){
if (left < 0){
right++;
} else if (right >= arr.length ){
left--;
} else if (x - arr[left] <= arr[right] - x){
left--;
} else {
right++;
}
}
List<Integer> list = new ArrayList<>();
for (int i = left + 1; i < right; i++){
list.add(arr[i]);
}
return list;
}
private int binary(int[] arr, int x){
int left = 0, right = arr.length - 1;
while (left < right){
int mid = (left + right) >> 1;
// 如果中位数大于x
if (arr[mid] >= x){
right = mid;
}else { // 如果中位数小于x
left = mid + 1;
}
}
return left;
}