LeetCode 腾讯50题Python实现之《最小栈》

题目

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

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

解题思路一

划重点：“常数时间内”；“一个栈”；
常数时间，意味着，不能用排序/查找方法来实现。那么就只能空间换时间，由于数据结构栈的严格严格“后入先出”顺序，当数字传入的时候，就把当时最小的索引保存下来。这样当此元素在栈顶时，就可以保证其对应的“传入时最小值“，就是此时栈中最小值。

代码

class MinStack(object):

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

    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        current_min = self.getMin()
        if not current_min or x < current_min:
            current_min = x
        self.my_stack.append((x, current_min))  
        # 因为栈的入栈、出栈顺序是严格的后入先出，这里保存每一个元素入栈时，当时的最小值，那么就可以保证，当
        # 此元素在栈顶时，其对应的current_min，就是当前栈中最小值

    def pop(self):
        """
        :rtype: None
        """
        self.my_stack.pop()

    def top(self):
        """
        :rtype: int
        """
        return self.my_stack[-1][0]

    def getMin(self):
        """
        :rtype: int
        """
        return None if len(self.my_stack) == 0 else self.my_stack[-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()

问题：此代码，有测试用例未通过：

输入：
["MinStack","push","push","push","getMin","pop","getMin"]
[[],[0],[1],[0],[],[],[]]
输出：
[null,null,null,null,0,null,1]
预期：
[null,null,null,null,0,null,0]

查看评论有一代码和我思路相似，但是更加简洁，并且完美AC

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]

作者：QQqun902025048
链接：https://leetcode-cn.com/problems/two-sum/solution/python-mei-ge-yi-xing-by-knifezhu/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

思路二

使用辅助栈，在辅助栈里对数据利用内置sorted函数排序，虽然在leetcode里也能过，但是感觉如果大的数据栈情况，这种方法可能效率有限吧