【20190709】【每天一道算法题】最小栈（栈）

问题

设计一个支持 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.

---

思路及解答

1. Python实现

# 方法：用两个栈
# 用两个栈，一个存放元素，另一个栈顶始终存放当前最小元素。注意：Stack 弹出元素时，要判断是不是当前最小元素，若是则 minStack 也要弹出栈顶。
# 时间复杂度：O(n)
class MinStack:
    def __init__(self):
        """
        initialize your data structure here.
        """
        self.Stack = []
        self.minStack = []

    def push(self, x: int) -> None:
        self.Stack.append(x)
        if not self.minStack or x <= self.minStack[-1]:   # 这里是 x<=，而不是 x<，不然会出错！
            self.minStack.append(x)

    def pop(self) -> None:
        if self.Stack.pop() == self.minStack[-1]:   # 如果弹出元素是最小元素，那么也应该把最小栈中的栈顶元素弹出
            self.minStack.pop()
        # self.Stack.pop()      # 这样不对

    def top(self) -> int:
        return self.Stack[-1]

    def getMin(self) -> int:
        return self.minStack[-1]