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

合集 - LeetCode 二分法(17)

1.35. 搜索插入位置2024-09-092.704. 二分查找2024-09-093.744. 寻找比目标字母大的最小字母2024-09-094.2529. 正整数和负整数的最大计数2024-09-095.1385. 两个数组间的距离值2024-09-096.2300. 咒语和药水的成功对数2024-09-097.2389. 和有限的最长子序列2024-09-098.1170. 比较字符串最小字母出现频次2024-09-099.2080. 区间内查询数字的频率2024-09-0910.2563. 统计公平数对的数目2024-09-0911.981. 基于时间的键值存储2024-09-1012.1146. 快照数组2024-09-10

13.658. 找到 K 个最接近的元素2024-09-10

14.1818. 绝对差值和2024-09-1015.911. 在线选举2024-09-1016.LCP 08. 剧情触发时间2024-09-1017.[模板题] - 300. 最长递增子序列2024-09-13

收起

题目链接	658. 找到 K 个最接近的元素
思路	二分+双指针
题解链接	官方题解
关键点	二分：查找到第一个>= val的位置；双指针：利用“有序”性质，双指针迭代跟踪left & right的位置
时间复杂度	$O(\log n+k)$
空间复杂度	$O(1)$

代码实现：

class Solution:
    def findClosestElements(self, arr: List[int], k: int, x: int) -> List[int]:
        left, right = -1, len(arr)
        while left + 1 < right:
            mid = (left+right) // 2
            if arr[mid] < x:
                left = mid
            else:
                right = mid
        left = right - 1
        for _ in range(k):
            if left < 0:
                right += 1
            elif right >= len(arr) or x - arr[left] <= arr[right] - x:
                left -= 1
            else:
                right += 1
        return arr[left+1:right]

Python-官方库

class Solution:
    def findClosestElements(self, arr: List[int], k: int, x: int) -> List[int]:
        right = bisect_left(arr, x)
        left = right - 1
        for _ in range(k):
            if left < 0:
                right += 1
            elif right >= len(arr) or x - arr[left] <= arr[right] - x:
                left -= 1
            else:
                right += 1
        return arr[left+1:right]