155. Min Stack

最新推荐文章于 2020-08-03 10:03:11 发布

原创最新推荐文章于 2020-08-03 10:03:11 发布 · 173 阅读

0 ·

CC 4.0 BY-SA版权

leetcode代码笔记专栏收录该内容

82 篇文章

订阅专栏

本文介绍了一种特殊的数据结构——最小栈，它能在常数时间内完成压栈、弹栈、获取栈顶元素及检索栈内最小元素的操作。通过两个实例展示了如何使用辅助栈或数组来维护栈内元素的最小值。

esign 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.

Example:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> Returns -3.
minStack.pop();
minStack.top();      --> Returns 0.
minStack.getMin();   --> Returns -2.

class MinStack {
public:
    vector<int> a;
    vector<int> min;
    MinStack() {
        min.push_back(2147483647);
    }
    void push(int x) {
        a.push_back(x);
        min.push_back(x < min.back() ? x : min.back());
    }

    void pop() {
        a.pop_back();
        min.pop_back();
    }

    int top() {
        return a.back();
    }

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

class MinStack {
public:
    stack<int> st;
    stack<int> nextMin;
    
    void push(int x) {
        if(st.empty()){
            nextMin.push(x);
        }else if(nextMin.top() >= x){
            nextMin.push(x);
        }
        st.push(x);
    }

    void pop() {
        int tmp = st.top();
        st.pop();
        if(tmp == nextMin.top()){
            nextMin.pop();
        }
    }

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

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