【LeetCode第295场周赛】C - 使数组按非递减顺序排列（单调栈解法详解）

这道题在赛时尝试了很多局部正确的做法，也就是贪心。现在想来这种题目大概就是一步步把每个细节都思考清楚，再通过不断尝试才可能通过的题？以上仅个人愚见。

题意：
给你一个下标从 $0$ 开始的整数数组 $nums$ 。在一步操作中，移除所有满足 $nums[i-1]>nums[i]$ 的 $nums[i]$ ，其中 $0<i<nums.length$ 。重复执行步骤，直到 $nums$ 变为 非递减 数组，返回所需执行的操作数。
数据范围：
$1\le nums.length\le 10^5$
$1\le nums[i]\le 10^9$

题解：
直观想法：

考虑每个元素被删除的时间点，元素 $nums[i]$ 如果可以被删除，一定是被它左边第一个比它大的元素 $nums[j],j<i$ 删除的，我们并不需要担心 $nums[j]$ 在与 $nums[i]$ 位置相邻之前就可能被 $nums[k],k<j$ 删除。如果这样，可以等价将 $nums[k]$ 删除 $nums[i]$ 看成 $nums[j]$ 删除 $nums[i]$。

$dp[i]$ 表示将前 $i$ 个元素按照题目所示变成非递减数组的最迟时间点。
初始 $dp[i]=0$
当存在 $nums[i]<nums[i-1]$ 时，$dp[i]=1$
否则找到第一个 $nums[j]$，存在$j<i$ 且 $nums[j]>nums[i]$ 。元素 $nums[j+1,...,i-1]$都不比 $nums[i]$ 大，只有这些元素被删除后，$nums[i]$才会被删除。
所以，$nums[i]$ 被删除的时间点等于 $nums[i+1,...,j-1]$ 这些元素中最大被删除的时间点 + 1

一种显然的策略是：$t[i]$ 表示第$i$ 个数被删除的时间点，类似 $dp$ 的思想， $O(n^2)$ 转移

class Solution {
public:
    int totalSteps(vector<int>& nums) {
       // t[i] 表示将第 i 个数 按照题目要求维护成一个非递减数组，需要的操作次数
        int n = nums.size();
		
        vector<int> t(n, 0);
        int maxv = nums[0];
        for(int i = 1; i < n; ++i) {
            int temp = 0;
            for(int j = i - 1; j >= 0; --j) {
                if(nums[j] > nums[i]) break;
                temp = max(temp, t[j]);
            }
            
        	if(nums[i] < maxv) t[i] = temp + 1;
            maxv = max(maxv, nums[i]);
		}
        
        int ans = 0;
        for(int i = 0; i < n; ++i)
            ans = max(ans, t[i]);
        
        return ans;
    }
};

显然时间复杂度是不够优秀的，再来思考如何优化第二层循环。

因为 $nums[i]$ 一定不小于 $nums[j+1,...,i-1]$，而 $t[i]$ 也一定比 $t[j+1,...,i-1]$大。
这样一个显然的策略是，维护一个单调递减的栈，栈中单调递减的为元素的值 $nums[i]$ ，同时存储这个元素被删除的时间点 $t[i]$ 。

这里使用单调递减的栈的原因是，之后的元素 $nums[k],k>i(nums[k]>nums[i])$， 并不由 $nums[i]$ 删除，而是由在栈中的大于 $nums[i]$ 的元素 $nums[j](j<i,nums[j]>nums[i],nums[j]>nums[k])$ 删除。但是，在 $nums[j]$ 删除 $nums[k]$ 前，必须先删除 $nums[j+1,...,k-1]$，因此，这些元素中最大被删除的时间点 + 1 即为$nums[k]$ 被删除的时间点。注意，如果 $nums[0,...,i-1]$ 都不大于 $nums[i]$ ，则这个元素不会被删除，其时间点设置为 $0$ 即可。

代码：

class Solution {
public:
    int totalSteps(vector<int>& nums) {
        int n = nums.size();
        stack<pair<int, int>> stk;
        
        int ans = 0;
        for(int i = 0; i < n; ++i) {
            int t = 0;
            while(!stk.empty() && stk.top().first <= nums[i]) {
                t = max(t, stk.top().second);
                stk.pop();
            }
            if(stk.empty()) t = 0; // 0~i-1中没有元素大于nums[i]
            else t += 1;
            stk.push({nums[i], t});
            
            ans = max(ans, t);
		}
        
        return ans;
    }
};