剑指 Offer 30. 包含min函数的栈

剑指 Offer 30. 包含min函数的栈

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例：
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
来源：力扣（LeetCode）

思路：维护两个栈a，b。a用来用于存储所有元素，保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常实现。b用来维护当前a中非严格降序的元素，即栈a的最小元素始终是栈b的栈顶元素。即维护好 栈 B 的元素，使其保持非严格降序，即可实现 min() 函数的 O(1)复杂度。
具体实现：
初始化两个栈a，b。使b初始化值为最大值。
push函数：将元素x加入到栈a中，此时如果栈b为空，判断x是否小于等于b的栈顶元素，若是，则将x加入到b中。
pop函数，为了维护保持栈 a，b的 元素一致性 。所以需要判断a，b的栈顶元素是否一致，一致的话则进行出栈操作。
top函数：直接返回a的栈顶元素。
min函数：直接返回b的栈顶元素。
具体代码如下：

class MinStack {
    stack<int> a,b;
public:
    /** initialize your data structure here. */
    MinStack() {
        b.push(INT_MAX);
    }
    
    void push(int x) {
        a.push(x);
        if(b.empty() || x <= b.top()) b.push(x);

    }
    
    void pop() {
        if( a.top() == b.top())
         b.pop();
         a.pop();

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

    }
    
    int min() {
        return b.top();

    }
};

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack* obj = new MinStack();
 * obj->push(x);
 * obj->pop();
 * int param_3 = obj->top();
 * int param_4 = obj->min();
 */