LeetCode：最小栈（Python版本）

最小栈

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/

设计一个支持 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(), pop(), top(), 这些操作很容易实现，用list即可, 但是题目要求getMin(), 需要在常数时间内得到最小值，这个就要简单思考一下，如果说从无序列表中得到最小值，那么时间复杂度将会是o(n)，不满足题目要求，所以我们需要在push()，操作时，把最小值保存下来。

Python代码

class MinStack(object):

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

    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        self.nums.append(x)
        self.minum.append(x)
        self.minum.sort(reverse=True)
        

    def pop(self):
        """
        :rtype: None
        """
        if self.nums.pop() == self.minum[-1]:
            self.minum.pop()
               
    def top(self):
        """
        :rtype: int
        """
        return self.nums[-1]

    def getMin(self):
        """
        :rtype: int
        """
        return self.minum[-1]
        
# 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()

执行用时 : 416 ms, 在Min Stack的Python提交中击败了43.19% 的用户
内存消耗 : 15.8 MB, 在Min Stack的Python提交中击败了5.01% 的用户

题解区发现了更简洁的写法，速度也更快。注意是Python3!–>传送门

class MinStack:
    
    def __init__(self):
        self.data = [(None, float('inf'))]

    def push(self, x: 'int') -> 'None':
        self.data.append((x, min(x, self.data[-1][1])))

    def pop(self) -> 'None':
        if len(self.data) > 1: self.data.pop()

    def top(self) -> 'int':
        return self.data[-1][0]

    def getMin(self) -> 'int':
        return self.data[-1][1]
        
# 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()

执行用时 : 72 ms, 在Min Stack的Python3提交中击败了95.12% 的用户
内存消耗 : 16.6 MB, 在Min Stack的Python3提交中击败了38.66% 的用户