【Leetcode】910. Smallest Range II（配数学证明）

题目地址：

https://leetcode.com/problems/smallest-range-ii/description/

给定一个长$n$数组$a$和一个非负整数$k$，强制让每一项减或者加$k$，要使得整个数组的极差最小，问最小的极差是多少。

假设已经从小到大排好序。然后，可以证明对于任意使得极差最小的方案，都可以将其调整为，有个分界线，之前都是$+k$后面都是$-k$的形式。如果不然，可以找到$i,a_i-k,a_{i+1}+k$，在这种方案下，最大值一定是$\ge a_{i+1}+k$，那么将$a_i-k$变成$a_i+k$的话，最大值不会变，而最小值有可能变大，所以极差不会变大。所以分界线存在。直接枚举分界线即可。如果分界线是$i$，即新数组为$$a_0+k,...,a_i+k,a_{i+1}-k,...,a_{n-1}-k$$这个数组的最小值和最大值很容易求。代码如下：

class Solution {
 public:
  int smallestRangeII(vector<int>& a, int k) {
    sort(a.begin(), a.end());
    int res = a.back() - a[0];
    for (int i = 0; i + 1 < a.size(); i++) {
      int M = max(a.back() - k, a[i] + k);
      int m = min(a[0] + k, a[i + 1] - k);
      res = min(res, M - m);
    }
    return res;
  }
};

时间复杂度$O(n\log n)$，空间$O(1)$。