smallestRangeI.py

# -*- encoding: utf-8 -*-
'''
@project :   LeetCode
@File    :   smallestRangeI.py
@Contact :   9824373@qq.com
@Desc    :
            给定一个整数数组 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]

@Modify Time                @Author     @Version    @Desciption
------------                -------     --------    -----------
2020-03-20     zhan        1.0         None
'''

'''
ans = (max(A) - K) - (min(A) + K)。
如果 ans > 0 则起就是答案
若 ans <= 0 ,则 0 
'''
from typing import List

class Solution:
    def smallestRangeI(self, A: List[int], K: int) -> int:
        # A.sort()
        min_v, max_v = min(A), max(A)
        ans = max_v - K - min_v - K
        if ans > 0: return ans
        else: return 0

if __name__ == '__main__':
    A = [1, 3, 6]
    K = 3

    ans = Solution().smallestRangeI(A, K)
    print(ans)