【LeetCode-155】最小栈

问题

设计一个支持 push ，pop ，top 操作，并能在常数时间内检索到最小元素的栈。

• push(x) —— 将元素 x 推入栈中。
• pop() —— 删除栈顶的元素。
• top() —— 获取栈顶元素。
• getMin() —— 检索栈中的最小元素。

解答1：辅助栈

class MinStack {
public:
    void push(int val) {
        s.push(val);
        if (s_min.empty() || s_min.top() >= val) s_min.push(val);
    }
    void pop() {
        if (!s_min.empty() && s_min.top() == s.top()) s_min.pop();
        s.pop();
    }
    int top() {
        return s.top();
    }
    int getMin() {
        return s_min.top();
    }
private:
    stack<int> s, s_min;
};

重点思路

思路和【剑指Offer-59-II】队列的最大值类似。

解答2：一个栈

class MinStack {
public:
    void push(int val) {
        if (min >= val) {
            s.push(min);
            min = val;
        }
        s.push(val);
    }
    void pop() {
        if (s.top() == min) {
            s.pop();
            min = s.top();
        }
        s.pop();
    }
    int top() {
        return s.top();
    }
    int getMin() {
        return min;
    }
private:
    stack<int> s;
    int min = INT_MAX;
};

重点思路

实际上塞进栈里的数据并没有少，但就是少了一个栈。主要的思路是遇到最小值时额外push一份最小值。