leetcode思路总结反思(stack)

leetcode思路总结反思(stack)

stack的常见思路归纳如下：

1）monotonous increase stack

基本的框架如下：

for(int i = 0; i < A.size(); i++){
  while(!in_stk.empty() && in_stk.top() > A[i]){
    in_stk.pop();
  }
  in_stk.push(A[i]);
}

monotonous increase stack的作用是在stack中维持非严格升序，可以解决的一般问题有：
(1) find the previous less element of each element in a vector with O(n) time:

for(int i = 0; i < A.size(); i++){
  while(!in_stk.empty() && A[in_stk.top()] > A[i]){
    in_stk.pop();
  }
  previous_less[i] = in_stk.empty()? -1: in_stk.top();
  in_stk.push(i);
}

(2) find the next less element of each element in a vector with O(n) time:

vector<int> previous_less(A.size(), -1);
for(int i = 0; i < A.size(); i++){
  while(!in_stk.empty() && A[in_stk.top()] > A[i]){
    auto x = in_stk.top(); in_stk.pop();
    next_less[x] = i;
  }
  in_stk.push(i);
}

2）

496. Next Greater Element I

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res(nums1.size(),-1);
        unordered_map<int,int> map_;
        
        stack<int> stk;
        for(int i=0;i<nums2.size();i++)
        {
            while(!stk.empty()&&nums2[i]>stk.top())
            {
                map_[stk.top()]=nums2[i];
                //res[stk.top()]=nums2[i];
                stk.pop();
            }
            stk.push(nums2[i]);
        }
        
        for(int i=0;i<nums1.size();i++)
        {
            if(map_.find(nums1[i])!=map_.end())
                res[i]=map_[nums1[i]];
        }
        return res;
    }
};

该题关键点在于用stack维护一个单调非严格递减的栈，每次遇到比栈顶元素大的值，便进行出栈赋值操作。遇到小于等于栈顶元素的值便压入栈。
类似题还有
503.Next Greater Element II
739. Daily Temperatures,
1019. Next Greater Node In Linked List

402. Remove K Digits

class Solution {
public:
    string removeKdigits(string num, int k) {
        string stk;
        for(int i=0;i<num.size();i++)
        {
            while(!stk.empty()&&stk.back()>num[i]&&k)
            {
                stk.pop_back();
                k--;
            }
            if(num[i]!='0'||!stk.empty())
                stk.push_back(num[i]);    
        }
        while((k--)&&!stk.empty())
            stk.pop_back();
        return stk.empty()?"0":stk;
    }
};

该题关键点在于用stack维护一个单调非严格递增的栈，为了最后方便输出，这里使用string,pop_back,push_back来代替stack，pop(),push()。这题有很多小地方需要注意，例如，前导零的判断，到最后如果k不为零那么还需要做pop()操作，最后如果是空string则输出“0”而不是输出“”等；