LeetCode

Min Stack

2015.1.23 12:13

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.

Solution:

　　This could almost be counted as a FAQ. You'll find a same question in "Cracking the Coding Interview". A min stack is maintained alongside the data stack, recording every minimal (so far) element. The min stack is updated whever a new minimal value is being pushed, or the current minimal value is being popped.

　　Total time complexity for each operation is strictly O(1). Extra space of O(n) is required to maintain the min stack.

Accepted code:

// 1AC, old
#include <stack>
using namespace std;

class MinStack {
public:
    void push(int x) {
        if (ms.empty() || ms.top() >= x) {
            ms.push(x);
        }
        s.push(x);
    }

    void pop() {
        if (s.top() == ms.top()) {
            ms.pop();
        }
        s.pop();
    }

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

    int getMin() {
        return ms.top();
    }
private:
    stack<int> s, ms;
};