给定一个排序好的数组,两个整数 k 和 x,从数组中找到最靠近 x(两数之差最小)的 k 个数。返回的结果必须要是按升序排好的。如果有两个数与 x 的差值一样,优先选择数值较小的那个数。
示例 1:
输入: [1,2,3,4,5], k=4, x=3
输出: [1,2,3,4]
示例 2:
输入: [1,2,3,4,5], k=4, x=-1
输出: [1,2,3,4]
说明:
1、k 的值为正数,且总是小于给定排序数组的长度。
2、数组不为空,且长度不超过 104
3、数组里的每个元素与 x 的绝对值不超过 104
解
双指针遍历数组,每次向中间移动距离较大的指针(相当于扔掉一个数据),指导大小满足要求。
public static List<Integer> findClosestElements(int[] arr, int k, int x) {
List<Integer> res = new ArrayList<Integer>();
int len = arr.length;
int i = 0, j = len - 1;
int pop = k;
while(i<=j&&(j-i+1)>pop) {
int left = Math.abs(arr[i] - x);
int right = Math.abs(arr[j] - x);
if(left<=right)
j--;
else
i++;
// System.out.println(i+":"+j);
}
for(;i<=j;i++)
res.add(arr[i]);
return res;
}