Question

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

English Version

题目描述

给你一个下标从 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

解法

方法一：单调栈

class Solution:
  def totalSteps(self, nums: List[int]) -> int:
    stk = []
    ans, n = 0, len(nums)
    dp = [0] * n
    for i in range(n - 1, -1, -1):
      while stk and nums[i] > nums[stk[-1]]:
        dp[i] = max(dp[i] + 1, dp[stk.pop()])
      stk.append(i)
    return max(dp)

class Solution {
  public int totalSteps(int[] nums) {
    Deque<Integer> stk = new ArrayDeque<>();
    int ans = 0;
    int n = nums.length;
    int[] dp = new int[n];
    for (int i = n - 1; i >= 0; --i) {
      while (!stk.isEmpty() && nums[i] > nums[stk.peek()]) {
        dp[i] = Math.max(dp[i] + 1, dp[stk.pop()]);
        ans = Math.max(ans, dp[i]);
      }
      stk.push(i);
    }
    return ans;
  }
}

class Solution {
public:
  int totalSteps(vector<int>& nums) {
    stack<int> stk;
    int ans = 0, n = nums.size();
    vector<int> dp(n);
    for (int i = n - 1; i >= 0; --i) {
      while (!stk.empty() && nums[i] > nums[stk.top()]) {
        dp[i] = max(dp[i] + 1, dp[stk.top()]);
        ans = max(ans, dp[i]);
        stk.pop();
      }
      stk.push(i);
    }
    return ans;
  }
};

func totalSteps(nums []int) int {
  stk := []int{}
  ans, n := 0, len(nums)
  dp := make([]int, n)
  for i := n - 1; i >= 0; i-- {
    for len(stk) > 0 && nums[i] > nums[stk[len(stk)-1]] {
      dp[i] = max(dp[i]+1, dp[stk[len(stk)-1]])
      stk = stk[:len(stk)-1]
      ans = max(ans, dp[i])
    }
    stk = append(stk, i)
  }
  return ans
}

function totalSteps(nums: number[]): number {
  let ans = 0;
  let stack = [];
  for (let num of nums) {
    let max = 0;
    while (stack.length && stack[0][0] <= num) {
      max = Math.max(stack[0][1], max);
      stack.shift();
    }
    if (stack.length) max++;
    ans = Math.max(max, ans);
    stack.unshift([num, max]);
  }
  return ans;
}