LeetCode题解——910. 最小差值 II

题目

https://leetcode-cn.com/problems/smallest-range-ii/
给定一个整数数组 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

题目看半天都没看懂…

思路1

递归调用，分别求每一个的差值，结果超时了。。。

代码

class Solution {
    public static int smallestRangeII(int[] A, int K) {

        return core(A, 0, K);
    }

    public static int core(int[] A, int index, int K) {
        if (index == A.length) {
            return dif(A);
        }
        A[index] += K;
        int a = core(A, index + 1, K);
        A[index] = A[index] - K * 2;
        int b = core(A, index + 1, K);
        A[index] += K;
        return a < b ? a : b;
    }

    public static int dif(int[] A) {
        int min = A[0];
        int max = A[0];
        for (int i = 0; i < A.length; i++) {
            min = Math.min(min, A[i]);
            max = Math.max(max, A[i]);
        }
        return max - min;
    }
}

思路2

看了评论看半天才想明白，数学题。
为了使得差值最小，那么先对A排序，假如处理到A[i]，那
A[i]，A[0,i] 所有数字 < A[i+1,len-1] 所有数字，设前面A1后面A2,
为了使整体最小，那必然有A1+K与A2-K才会最小。
那最大值一定是：A[i]+K或者A[len-1]-K
那最小值一定是：A[0]+K或者A[i+1]-K
做差值保存最小值，遍历下一个节点就行。

代码

import java.util.Arrays;

public class Solution_910 {
    public static void main(String[] args) {
        System.out.println(smallestRangeII(new int[]{2, 7, 2}, 1));
    }

    public static int smallestRangeII(int[] A, int K) {
        Arrays.sort(A);
        int n = A.length;
        int res = A[n - 1] - A[0];
        for (int i = 0; i < n-1; i++) {
            int min = Math.min(A[0] + K, A[i+1] - K);
            int max = Math.max(A[n - 1] - K, A[i] + K);
            res = Math.min(max - min, res);
        }
        return res;
    }

}