1. 程式人生 > >leetcode 908. 最小差值 I

leetcode 908. 最小差值 I

給定一個整數陣列 A,對於每個整數 A[i],我們可以選擇任意 x 滿足 -K <= x <= K,並將 x 加到 A[i] 中。

在此過程之後,我們得到一些陣列 B

返回 B 的最大值和 B 的最小值之間可能存在的最小差值。

 

示例 1:

輸入:A = [1], K = 0
輸出:0
解釋:B = [1]

示例 2:

輸入:A = [0,10], K = 2
輸出:6
解釋:B = [2,8]

示例 3:

輸入:A = [1,3,6], K = 3
輸出:0
解釋:B = [3,3,3] 或 B = [4,4,4]

 

提示:

  1. 1 <= A.length <= 10000
  2. 0 <= A[i] <= 10000
  3. 0 <= K <= 10000

思路:這題中心思想就是“補齊短板”,只需要盯住最大最小值就行。分兩種情況:1)max - min >= 2K;2)max - min <2K

class Solution(object):
    def smallestRangeI(self, A, K):
        """
        :type A: List[int]
        :type K: int
        :rtype: int
        """
        if len(A) == 1:
            return 0
        res = max(A) - min(A)
        if res >= 2*K:
            return res -2*K
        else:
            return 0