力扣 658. 找到 K 个最接近的元素

658. 找到 K 个最接近的元素

难度中等84收藏分享切换为英文关注反馈

给定一个排序好的数组，两个整数 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]

 

解法：首先数组事有序的，如果x>arr[arr.lenght-1],那么倒数前k就是我们要求得。如果x<arr[0],那么正数前k个就是我们要求的。

那如果都不是呢，那就二分一下，找到数组里面嘴接近x的值。然后从这个中心扩散出一个范围。在这个范围在收缩，知道r-l<=k-1

class Solution {
    public List<Integer> findClosestElements(int[] arr, int k, int x) {
        List<Integer> list = Arrays.stream(arr).boxed().collect(Collectors.toList());
        int len=list.size();
        if(x<=list.get(0)){
            return list.subList(0,k);
        }
        else if(x>=list.get(len-1)){
            return  list.subList(len-k,len);
        }
        else{
            int index= Collections.binarySearch(list,x);
            if(index<0)
                index=-index-1;
            int l=Math.max(0,index-k-1);
            int r=Math.min(len-1,index+k-1);
            //System.out.println("index"+index+"    "+l+" "+r);
            while (r-l>k-1){
                if(x-list.get(l)>list.get(r)-x){
                    l++;
                }
                else if(x-list.get(l)<=list.get(r)-x){
                    r--;
                }
            }
            return list.subList(l,r+1);
        }
    }
}