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

给定一个 排序好 的数组 arr ，两个整数 k 和 x ，从数组中找到最靠近 x（两数之差最小）的 k 个数。返回的结果必须要是按升序排好的。

整数 a 比整数 b 更接近 x 需要满足：

• |a - x| < |b - x| 或者
• |a - x| == |b - x| 且 a < b

 

示例 1：

输入：arr = [1,2,3,4,5], k = 4, x = 3
输出：[1,2,3,4]

示例 2：

输入：arr = [1,2,3,4,5], k = 4, x = -1
输出：[1,2,3,4]

 

提示：

• 1 <= k <= arr.length
• 1 <= arr.length <= 104
• arr 按 升序 排列
• -104 <= arr[i], x <= 104

思路：

这道题的核心就是找到答案区间的左端点，因为长度已知是 k， 找到左端点后就可以确定答案区间。

每次二分我们都是在判断从 arr[mid] 和 arr[mid + k] 之间取哪一个，

因为它俩相隔 k + 1，所以两个中只可以取一个。

我们会取离 x 更近的那个值，然后将区间减半。

时间复杂度：O（logN）

空间复杂度：O（1）

class Solution:
    def findClosestElements(self, arr: List[int], k: int, x: int) -> List[int]:
        # find the left point of res
        n = len(arr)
        left, right = 0, n - k - 1
        while left <= right:
            mid = (left + right) // 2
            # arr[mid], x, arr[mid + k], only pick 1 from these two
            # if arr[mid] and x are closer, we go with arr[mid]
            if x - arr[mid] <= arr[mid + k] - x:
                right = mid - 1
            else:
                left = mid + 1
        return arr[left:left + k]