【Leetcode】【Easy】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.

实现一个栈的基本功能，并且可以用o(1)的时间取出栈的最小元素。

使用两个栈的解题方法：

建立两个栈容器，一个常规使用elementStack，一个保存栈内最小元素minStack。

注意：

常规栈中最小的元素可能不唯一。当某个最小元素出栈时，minStack也需做出出栈动作。因此minStack栈中记录最小元素的数量应该和elementStack中最小元素数量相同。

也就是，每次进行常规栈压栈时，都进行和min栈记录的最小元素比较，将小于等于min栈最小元素的常规元素压栈。

代码：

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

    void pop() {
        if (!element.empty()) {
            if (element.top() == min.top()) 
                min.pop();
            element.pop();
        }
    }

    int top() {
        if (!element.empty())
            return element.top();
    }

    int getMin() {
        if (!min.empty())
            return min.top();
    }
    
private:
    stack<int> element;
    stack<int> min;
};

也可以只使用一个栈，即常规栈。只是压栈的每个元素都包含当前栈内最小元素值。

即，假设当前栈内最小值为m，当对元素A压栈时，若A<=m，则压入 (A, A)，若A>m，则压入 (A, m)。

（代码略）

附录：

各种数据结构最简洁表示方法，参考《数据结构与算法分析—C语言版》