[LeetCode] 155_最小栈

[LeetCode] 155_最小栈

题目要求

设计一个支持 push，pop，top 操作，并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

题目分析

这道题是典型的用空间换时间，因为最小值需要多次取，如果每次都在取时重新进行查找，速度太慢了。这里使用一个额外的栈，每当push一个数入栈时，都在最小栈中放入push这个数之后，栈中的最小元素（通过对比此时的栈顶是否为空或元素大小来确定）。
所以当退栈时，最小栈顶的数也退出，这样可以保证最小栈的栈顶元素一定是当前所有元素中最小的。

注意点

在判断当前元素是不是最小的元素时，要注意先判断最小栈是否为空，如果为空，那么当前元素一定是最小元素，如果不为空，再比较栈顶元素和当前元素的大小。

代码

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