[LeetCode]155. 最小栈

题目

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

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

示例:

输入：
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]

输出：
[null,null,null,null,-3,null,0,-2]

解释：
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

提示：

• pop、top 和 getMin 操作总是在非空栈上调用。

解题思路

创建一个辅助栈sup，来保存原始栈中的最小元素。如果sup为空，或者sup的栈顶元素大于等于当前元素value，就把value加入sup。这样保证了sup栈中从栈底到栈顶是单调递减的，且sup的栈顶元素始终是原始栈中的最小元素。
具体步骤：
1）初始化原始栈stack，辅助栈supstack。
2）入栈：首先，将当前元素加入到原始栈中。其次，对于辅助栈，如果sup为空，或者sup的栈顶元素大于等于当前元素value，就把value加入sup。这样保证了sup栈中从栈底到栈顶是单调递减的，且sup的栈顶元素始终是原始栈中的最小元素。
3）出栈：原始栈弹出栈顶元素，将其与辅助栈的栈顶元素比较，如果相同，辅助栈的栈顶元素也出栈，否则，只有原始栈中的元素需要出栈。
4）取栈顶元素：即原始栈的栈顶元素。
5）求最小值：即辅助栈的栈顶元素。
注：Python中的list形式，用pop()的方法默认可以弹出list的最后一个元素，时间复杂度是O(1)。pop(k)表示弹出第k个元素，时间复杂度为O(k)。最坏情况下为O(n)。

复杂度分析：
时间复杂度：O(1)，入栈、出栈、取栈顶元素的时间复杂度都是O(1)。
空间复杂度：O(N)，需要用栈存储元素。

本题思想与（面试题59 - II）队列的最大值 有异曲同工之妙。

代码

Python代码如下：

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.stack = []
        self.supstack = []


    def push(self, x: int) -> None:
        self.stack.append(x)
        if(not self.supstack or self.supstack[-1]>=x):
            self.supstack.append(x)


    def pop(self) -> None:
        res = self.stack.pop()
        if res == self.supstack[-1]:
            self.supstack.pop()


    def top(self) -> int:
        if self.stack:
            res = self.stack[-1]
            return res

    def getMin(self) -> int:
        if self.supstack:
            min_num = self.supstack[-1]
            return min_num



# Your MinStack object will be instantiated and called as such:
# obj = MinStack()
# obj.push(x)
# obj.pop()
# param_3 = obj.top()
# param_4 = obj.getMin()

Java代码如下：

class MinStack {
    Stack<Integer> stack;
    Stack<Integer> supstack;

    /** initialize your data structure here. */
    public MinStack() {
        stack = new Stack<Integer>();
        supstack = new Stack<Integer>();
    }
    
    public void push(int x) {
        stack.add(x);
        if(supstack.isEmpty() || supstack.peek()>=x){
            supstack.add(x);
        }
    }
    
    public void pop() {
        int res = stack.pop();
        if(res == supstack.peek()){
            supstack.pop();
        }
    }
    
    public int top() {
        if(!stack.isEmpty()){
            int res = stack.peek();
            return res;
        }
        throw new RuntimeException("栈中元素为空，此操作非法");
    }
    
    public int getMin() {
        if(!supstack.isEmpty()){
            int min_num = supstack.peek();
            return min_num;
        }
        throw new RuntimeException("栈中元素为空，此操作非法");
    }
}

/**
 * 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.getMin();
 */