https://leetcode.cn/problems/find-k-closest-elements/
给定一个 排序好 的数组 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]
class Solution:
def findClosestElements(self, arr: List[int], k: int, x: int) -> List[int]:
# 数组、二分查找、库函数 bisect
from bisect import bisect_left
right = bisect_left(arr, x) # 如将x插入arr中,插在哪里,最小是0,最大是len(arr)
left = right - 1 # 找到在arr中离x最近的,左边的那个数的索引,最小是-1,最大是len(arr) - 1
print(left, right)
for _ in range(k):
# 如果 left 或 right 已经越界,那么不考虑对应部分的元素。
if left < 0:
right += 1 # left 越界, 最小是-1,此时不断向右扩展边界
elif right >= len(arr):
left -= 1 # right 越界,最大就是len(arr),不断调整左边界
elif x - arr[left] <= arr[right] - x:
# x - arr[left] <= arr[right] - x 说明arr[left]离x更近,宜往左移动,否则往右移动
left -= 1
else:
right += 1
return arr[left + 1: right]