658. K個の最も近い要素を見つける(中)

タイトル説明:

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

 

おすすめ

転載: blog.csdn.net/weixin_38664232/article/details/106480068