找到K個最接近的元素

題目描述:

解決代碼:
 

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

提交結果:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章