Leetcode 1509. 三次操作后最大值与最小值的最小差
给你一个数组 nums ,每次操作你可以选择 nums 中的任意一个元素并将它改成任意值。
请你返回三次操作后, nums 中最大值与最小值的差的最小值。
示例 1:
输入:nums = [5,3,2,4]
输出:0
解释:将数组 [5,3,2,4] 变成 [2,2,2,2].
最大值与最小值的差为 2-2 = 0 。
示例 2:
输入:nums = [1,5,0,10,14]
输出:1
解释:将数组 [1,5,0,10,14] 变成 [1,1,0,1,1] 。
最大值与最小值的差为 1-0 = 1 。
示例 3:
输入:nums = [6,6,0,1,1,4,6]
输出:2
示例 4:
输入:nums = [1,5,6,14,15]
输出:1
提示:
- 1 <= nums.length <= 10^5
- -10^ 9 <= nums[i] <= 10^9
主要思路:
先将数组排序,然后讨论一下四种情况
1.去除前三个元素
2.去除最后三个元素
3.去除前一个和后面最后两个
4.去除前二个和后面最后一个
Code:
class Solution {
public:
int minDifference(vector<int>& nums) {
if(nums.size()<=4)
return 0;
sort(nums.begin(),nums.end());
int a=*max_element(nums.begin(),nums.begin()+nums.size()-3);
int b=*min_element(nums.begin(),nums.begin()+3);
int res=a-b;
int a2=*max_element(nums.begin()+3,nums.end());
int b2=*min_element(nums.begin()+3,nums.end());
res=min(res,a2-b2);
int a3=*max_element(nums.begin()+2,nums.begin()+nums.size()-1);
int b3=*min_element(nums.begin()+2,nums.begin()+nums.size()-1);
res=min(res,a3-b3);
int a4=*max_element(nums.begin()+1,nums.begin()+nums.size()-2);
int b4=*min_element(nums.begin()+1,nums.begin()+nums.size()-2);
return min(res,a4-b4);
}
};