提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
文章目录
前言
给定一个 排序好 的数组 arr ,两个整数 k 和 x ,从数组中找到最靠近 x(两数之差最小)的 k 个数。返回的结果必须要是按升序排好的。整数 a 比整数 b 更接近 x 需要满足:
|a - x| < |b - x| 或者
|a - x| == |b - x| 且 a < b
思路:题目没说初始数组是降序升序,我们把x加入数组并升序排序,第一个与x值相同的元素默认为x,x左边第一个数为左哨兵left_x,右边第一个数为右哨兵right_x,当左哨兵索引>=0且右哨兵索引<=数组索引最大值时,比较两者对应的与x差的绝对值大小,左小则元素添加进result左哨兵左移一步,右小则元素添加进result右哨兵右移一步。总共循环k次。
代码如下(python):
class Solution:
def findClosestElements(self,arr,k,x):
arr.append(x)
arr.sort()
x_index = arr.index(x)
print(x_index)
#以x_index为中心点左右比较数组元素直到有k个为止
left_x = x_index-1
right_x = x_index+1
result = []
for i in range(k):
if left_x >= 0 and right <= len(arr)-1:
if abs(arr[left_x]-x) <= abs(arr[right_x]-x):
result.append(arr[left_x])
left_x -= 1
elif right_x <= len(arr)-1:
result.append(arr[right_x])
right_x += 1
elif right_x <= len(arr)-1:
result.append(arr[right_x])
right_x += 1
result.sort()
return result
a = Solution()
arr = list(map(int,input().split(',')))
k,x = map(int,input().split())
print(a.findClosestElements(arr,k,x))
结果:(注意我这里arr数组按逗号隔开,k与x按空格隔开)