タイトル説明:
给定一个排序好的数组,两个整数 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がmidとmid + kの間にあるような位置を見つけます。
class Solution:
def findClosestElements(self, arr: List[int], k: int, x: int) -> List[int]:
l = 0
r = len(arr) - k
while l < r:
mid = (r - l) // 2 + l
if x - arr[mid] > arr[mid + k] - x:
l = mid + 1
else:
r = mid
return arr[l: l + k]