【LeetCode】155. Min Stack

Min Stack

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.

 

相比传统stack（记为stk），为了记录最小值，需要再开一个最小值栈min。

min的栈顶top()始终为当前栈中的最小值。

class MinStack {
public:
    stack<int> stk;
    stack<int> min;
    
    void push(int x) {
        stk.push(x);
        if(min.empty() || x <= min.top())
            min.push(x);
    }

    void pop() {
        if(stk.top() == min.top())
        {
            stk.pop();
            min.pop();
        }
        else
            stk.pop();
    }

    int top() {
        return stk.top();
    }

    int getMin() {
        return min.top();
    }
};