leetcode-最小栈

题目：
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.

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.

这道题要求我们设计一个数据结构，其应该具有如下特性
1、在pop，push，和top方法上行为与stack一致
2、getMin方法可取得当前数据结构中最小的元素。

解题思路为，如果需要随时取得数据结构中最小元素，那么意味着我需要储存数据结构中最小的元素，而当这个最小元素被pop的时候，我应该能够得到下一个最小元素。
于是我们将使用两个栈来完成这件事。
第一个栈，就是普通的stack，称为sta
第二个栈，用于储存最小的元素，称为staMin

具体方法是
1、在push的时候，将元素与staMin的栈顶比较，若元素小于staMin.top，在push入sta的时候同时push入staMin。
2、pop时，看要被pop的元素是否与staMin.top相等，如果相等，两个栈同时pop。
3、top时，直接返回sta.top
4、getMin时，直接返回staMin.top

---

代码如下： 语言为cpp

class MinStack {
public:
    /** initialize your data structure here. */
    MinStack() {}
    
    void push(int x) {
        s1.push(x);
        if (s2.empty() || x <= s2.top()) s2.push(x);
    }
    
    void pop() {
        if (s1.top() == s2.top()) s2.pop();
        s1.pop();
    }
    
    int top() {
        return s1.top();
    }
    
    int getMin() {
        return s2.top();
    }
    
private:
    stack<int> s1, s2;
};