leetcode 910. 最小差值 II（C++、python）

给定一个整数数组 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
输出：3
解释：B = [4,6,3]

 

提示：

1 <= A.length <= 10000

0 <= A[i] <= 10000

0 <= K <= 10000

C++

class Solution {
public:
    int smallestRangeII(vector<int>& A, int K) 
    {
        int n=A.size();
        sort(A.begin(),A.end());
        int res=A[n-1]-A[0];
        for(int i=1;i<n;i++)
        {
            int small=min(A[0]+K,A[i]-K);
            int large=max(A[i-1]+K,A[n-1]-K);
            res=min(res,large-small);
        }
        return res;
    }
};

python

class Solution:
    def smallestRangeII(self, A: List[int], K: int) -> int:
        n=len(A)
        A=sorted(A)
        res=A[n-1]-A[0]
        for i in range(1,n):
            small=min(A[0]+K,A[i]-K)
            large=max(A[i-1]+K,A[n-1]-K)
            res=min(res,large-small)
        return res