題目描述:
解決代碼:
class Solution(object):
def findClosestElements(self, arr, k, x):
"""
:type arr: List[int]
:type k: int
:type x: int
:rtype: List[int]
"""
res = []
s = 0
for i in range(k):
s += abs(arr[i] - x)
small = s
res = arr[:k]
for i in range(k,len(arr)):
s += abs(arr[i]-x) - abs(arr[i-k] - x)
if s < small:
small = s
res = res[1:] + [arr[i]]
return res
提交結果: