【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();
    }
};