题目地址:
https://leetcode.com/problems/minimum-difference-between-largest-and-smallest-value-in-three-moves/
给定一个数组 A A A,允许修改其中三个数,问修改完后能得到最小的数组最大最小值差是多少。
首先,如果数组长度不超过 4 4 4,那就将三个数都改成第四个数即可,得到的差是 0 0 0,一定是最小的。如果长度超过 4 4 4,先对数组排序,我们可以证明,最优解一定是在首尾的地方改掉 3 3 3个元素,也就是最终答案一定是 A [ l A − 4 ] − A [ 0 ] , A [ l A − 3 ] − A [ 1 ] , A [ l A − 2 ] − A [ 2 ] , A [ l A − 1 ] − A [ 3 ] A[l_A-4]-A[0],A[l_A-3]-A[1],A[l_A-2]-A[2],A[l_A-1]-A[3] A[lA−4]−A[0],A[lA−3]−A[1],A[lA−2]−A[2],A[lA−1]−A[3]这几个其中的一个。可以用反证法。显然最优解一定不超过上面的几个可能的解,因为上面的解都是可行解,可以构造得到。如果最优解比上面几个可能的解都要严格小,那么这个最优解对应的最大值最小值之外一定有超过 3 3 3个元素,否则它就要被枚举到,矛盾。代码如下:
import java.util.Arrays;
public class Solution {
public int minDifference(int[] nums) {
if (nums.length <= 4) {
return 0;
}
Arrays.sort(nums);
int res = Integer.MAX_VALUE;
for (int i = 0; i <= 3; i++) {
res = Math.min(res, nums[nums.length - 1 - i] - nums[3 - i]);
}
return res;
}
}
时间复杂度 O ( n log n ) O(n\log n) O(nlogn),空间 O ( 1 ) O(1) O(1)。