给你一个下标从 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^5
  • nums 中的整数 互不相同

分析:由于 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);
    }
};

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐