Leetcode 2289. Steps to Make Array Non-decreasing (單調棧)

題目

題解：

• 對於每個元素，它只有當左邊的元素大於它的時候才能被刪去。
• 維護一個數組，dp dp[i]表示第i個元素被刪去需要多少步
• 維護一個單調遞減的棧，用來查找每個元素的左邊的第一個大於它的元素的位置
• 對於每個元素i，假設左邊第一個大於的元素的位置是k,那麼元素i需要等待區間[i-k]的元素都被刪去纔會達成讓元素i刪除的條件。注意這個時候元素i的左邊第一個元素不一定是k，但是無論是不是k，元素i都滿足被刪除的條件。所以對於每個元素i，只需要求出在區間[i-k]的dp數組的最大值，加上1就是刪除元素i所需要的步數。

class Solution {
public:
    
    int s[100005];
    int dp[100005];
    int totalSteps(vector<int>& nums) {
     
        int ans =0;
        s[0] = 0;
        dp[0]=0;
        int pos =0;
        for(int i=1;i<nums.size();i++)
        {
            if(nums[i]<nums[s[pos]])
            {
                ans = max(ans, 1);
                dp[i] = 1;
                s[++pos] = i;
            }
            else
            {
                int x = 0;
                while(pos!= -1 && nums[i]>=nums[s[pos]])
                {
                    x = max(x, dp[s[pos]]); 
                    pos--;          
                }
                
                if(pos == -1)
                {
                    dp[i] = 0;
                }
                else
                {
                    dp[i] = x + 1;
                    ans = max(ans, dp[i]);
                }
                
                s[++pos] = i; 
            }
            
        }
        return ans;
    }
};