[leetcode]658. Find K Closest Elements绝对距离最近的K个元素

Given a sorted array, two integers k and x, find the k closest elements to x in the array. The result should also be sorted in ascending order. If there is a tie, the smaller elements are always preferred.

Example 1:

Input: [1,2,3,4,5], k=4, x=3
Output: [1,2,3,4]

Example 2:

Input: [1,2,3,4,5], k=4, x=-1
Output: [1,2,3,4]

Note:

1. The value k is positive and will always be smaller than the length of the sorted array.
2. Length of the given array is positive and will not exceed 104
3. Absolute value of elements in the array and x will not exceed 104

题目

思路

本题要求我们在sorted array中找出K个最相近于x的数。因为输出结果一定排好序的、k-size的区间，若能找出该区间的leftBound，向右边走k个值，就可以得到desired output。

即问题转化为，怎样找到该leftBound呢？ 在[0, n-k]中使用binary search查找

代码

class Solution {
  public List<Integer> findClosestElements(int[] arr, int k, int x) {
       int begin = 0;
       int end = arr.length - k;
       while(begin < end){
               int mid = begin + (end - begin) /2 ; 
               if(x > arr[mid]){
                   if( x - arr[mid] > arr[mid + k] - x){
                       begin = mid +1;
                   }else{
                       end = mid;
                   }
               }else{
                   end = mid;
               }
       }
       int index = begin;
       List<Integer> result = new ArrayList<>() ;
       while( k != 0){
               result.add(arr[index++]);
               k--;
       }
       return result;
    }
}