1. 程式人生 > >【leetcode】910. Smallest Range II

【leetcode】910. Smallest Range II

turn lis bject ges obj 找出最大值 elif 特殊情況 sel

題目如下:

技術分享圖片

解題思路:我的思路是先找出最大值。對於數組中任意一個元素A[i]來說,如果A[i] + K 是B中的最大值,那麽意味著從A[i+1]開始的元素都要減去K,即如果有A[i] + K >= A[-1] - K,那麽A[i] + K 就可以作為最大值而存在;如果A[i] + K是最大值,那麽最大的最小值是多少呢?因為A[i] 左邊的元素都比A[i]小,所以其左邊的元素都可以加上K,最大的最小值就會在 A[0] + K 和 A[i+1] - K 之間產生。遍歷數組,計算每一個A[i] + K 作為最大值的時候最大值和最小值的差值即可。這裏還需要考慮一個特殊情況,就是A[-1] - K 作為最大值。如果是這樣情況,再遍歷一次數組,如果A[i] + K <= A[-1] -K,那麽最小值就可能是A[i] + K;如果A[i] + K > A[-1] -K,那麽最小值可能是A[i] - K 。

代碼如下:

class Solution(object):
    def smallestRangeII(self, A, K):
        """
        :type A: List[int]
        :type K: int
        :rtype: int
        """
        A.sort()
        res = 20000
        for i in range(len(A)):
            # suppose A[i] + K is largest
            if i == len(A) - 1:
                res 
= min(res,A[i] - A[0]) elif A[i] + K > A[-1] - K: diff = A[i] + K - min(A[i+1] - K,A[0] + K) res = min(res,diff) #A[-1] - K is largest maxv = A[-1] - K minv = A[0] + K for i in range(len(A)-1): if A[i] + K <= maxv:
continue else: minv = min(A[i] - K ,minv) res = min(res,maxv-minv) return res if res != 20000 else 0

【leetcode】910. Smallest Range II