5940. 從數組中移除最大值和最小值
給你一個下標從 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 <= 10510^5105
- ?105-10^5?105 <= nums[i] <= 10510^5105
- nums 中的整數 互不相同
解題思路
- 統計最大值和最小值所在的下標,因為在這題中,哪個是最大值,哪個是最小值對題目并不影響,因此我們只需要知道左邊極值的下標和右邊極值的下標即可。
- 共有三種刪除方法
- 從數組的 前面 移除元素,直到把右邊的極值移除
- 從數組的 后面 移除元素,直到把左邊的極值移除
- 從數組的兩邊一齊移除元素,直到把左邊和右邊的極值移除
代碼
class Solution {
public:int minimumDeletions(vector<int>& nums) {int max_i(0),min_i(0),n=nums.size();for (int i = 1; i < nums.size(); ++i) {if (nums[i]>nums[max_i])max_i=i;if (nums[i]<nums[min_i])min_i=i;}int l=min(max_i,min_i),r=max(max_i,min_i);return min(l+1+n-r,min(r+1,n-l));}
};