658. Encuentra K elementos más cercanos (medio)

Descripción del Título:

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

Idea: Redefina el objetivo de la dicotomía y encuentre una posición para que x esté entre mid y 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]

 

Supongo que te gusta

Origin blog.csdn.net/weixin_38664232/article/details/106480068
Recomendado
Clasificación