leetcode 2091. 从数组中移除最大值和最小值 中等
·
给你一个下标从 0 开始的数组 nums ,数组由若干 互不相同 的整数组成。
nums 中有一个值最小的元素和一个值最大的元素。分别称为 最小值 和 最大值 。你的目标是从数组中移除这两个元素。
一次 删除 操作定义为从数组的 前面 移除一个元素或从数组的 后面 移除一个元素。
返回将数组中最小值和最大值 都 移除需要的最小删除次数。
示例 1:
输入:nums = [2,10,7,5,4,1,8,6] 输出:5 解释: 数组中的最小元素是 nums[5] ,值为 1 。 数组中的最大元素是 nums[1] ,值为 10 。 将最大值和最小值都移除需要从数组前面移除 2 个元素,从数组后面移除 3 个元素。 结果是 2 + 3 = 5 ,这是所有可能情况中的最小删除次数。
示例 2:
输入:nums = [0,-4,19,1,8,-2,-3,5] 输出:3 解释: 数组中的最小元素是 nums[1] ,值为 -4 。 数组中的最大元素是 nums[2] ,值为 19 。 将最大值和最小值都移除需要从数组前面移除 3 个元素。 结果是 3 ,这是所有可能情况中的最小删除次数。
示例 3:
输入:nums = [101] 输出:1 解释: 数组中只有这一个元素,那么它既是数组中的最小值又是数组中的最大值。 移除它只需要 1 次删除操作。
提示:
1 <= nums.length <= 10^5-10^5 <= nums[i] <= 10^5nums中的整数 互不相同
分析:由于 nums 中的整数都互不相同,因此只有一个最大值和最小值,先遍历数组,分别找到最大值和最小值的位置。因为只要计算移除次数,所以不需要知道最大值和最小值哪个在前哪个在后,不妨设两个最值的位置为 max_pos 和 min_pos,其中 max_pos > min_pos。
移除方式总共有三种:1、移除掉 [0,max_pos];2、移除掉 [min_pos,n);3、移除掉 [0,min_pos] 和 [max_pos,n)。分别计算三种移除方式需要的操作次数,取较小值即可。
class Solution {
public:
int minimumDeletions(vector<int>& nums) {
int max_val,max_pos,min_val,min_pos,n=nums.size();
max_val=min_val=nums[0],max_pos=min_pos=0;
for(int i=1;i<n;++i)
{
if(nums[i]>max_val)max_val=nums[i],max_pos=i;
if(nums[i]<min_val)min_val=nums[i],min_pos=i;
}
if(max_pos<min_pos)swap(max_pos,min_pos);
return min(min(max_pos+1,n-min_pos),min_pos+1+n-max_pos);
}
};
更多推荐
所有评论(0)