单调栈2289. 使数组按非递减顺序排列

2022-06-11 21:17:00

问题描述：

给你一个下标从 0 开始的整数数组 nums 。在一步操作中，移除所有满足 nums[i - 1] > nums[i] 的 nums[i] ，其中 0 < i < nums.length 。

重复执行步骤，直到 nums 变为 非递减 数组，返回所需执行的操作数。

示例 1：

输入：nums = [5,3,4,4,7,3,6,11,8,5,11]
输出：3
解释：执行下述几个步骤：
- 步骤 1 ：[5,3,4,4,7,3,6,11,8,5,11] 变为 [5,4,4,7,6,11,11]
- 步骤 2 ：[5,4,4,7,6,11,11] 变为 [5,4,7,11,11]
- 步骤 3 ：[5,4,7,11,11] 变为 [5,7,11,11]
[5,7,11,11] 是一个非递减数组，因此，返回 3 。
示例 2：

输入：nums = [4,5,7,7,13]
输出：0
解释：nums 已经是一个非递减数组，因此，返回 0 。
 

提示：

1 <= nums.length <= 105
1 <= nums[i] <= 109

问题求解：

单调栈维护数字以及对应的删除时间，如果从来没有删除过则时间为0。

class Solution:
    def totalSteps(self, nums: List[int]) -> int:
        res = 0
        st = []
        for num in nums:
            curr = 0
            while st and st[-1][0] <= num:
                curr = max(curr, st.pop()[1])
            if st:
                curr += 1
            st.append([num, curr])
            res = max(res, curr)
        return res