[LeetCode]96. Min Stack带Min函数的栈

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

• push(x) -- Push element x onto stack.
• pop() -- Removes the element on top of the stack.
• top() -- Get the top element.
• getMin() -- Retrieve the minimum element in the stack.

 

Subscribe to see which companies asked this question

 

解法：还是用两个栈s1和s2来模拟。s1按顺序保存数据，而s2保存当前最小值。在进栈时，如果s2为空或者当前进栈值不大于s2栈顶值，则也进s2；出栈时，如果当前出栈值恰好为s2的栈顶值，则s2也要出栈。

 

class MinStack {
public:
    void push(int x) {
        s1.push(x);
        if (s2.empty() || x <= s2.top()) s2.push(x);
    }
    void pop() {
        if (!s1.empty()) {
            if (!s2.empty() && s1.top() == s2.top()) s2.pop();
            s1.pop();
        }
    }
    int top() {
        if (!s1.empty()) return s1.top();
        return 0;
    }
    int getMin() {
        if (!s2.empty()) return s2.top();
        return 0;
    }
private:
    std::stack<int> s1, s2;
};