LeetCode 算法题库【155】——最小栈

最小栈

题目描述：

解题思路：

• 第一种：辅助栈法。这个方法就是按照给的要求，我们先定义两个栈stack和min_result，然后一个一个函数来编写。先是push函数，这里我们要做的就是将元素x放入栈stack中，但是我们还要考虑的是还有一个函数时需要我们返回最小元素的，所以在这里要做上铺垫，我们将该元素放入查找最小元素的栈min_result中，而且为了找到最小的，我们还要做比较，将最小的那个元素放在栈顶，也就是min_result[-1]的位置。下一步我们编写pop函数，这个就没什么技巧，我们就把栈顶元素pop出来就可以了，只是还需要考虑的是这个栈顶的元素是不是最小元素，如果是的话，因为后面的运算就不会考虑到它了，那么也要在最小元素栈中将它除去。最后的top和getMin函数就更好写了，将所需要返回的返回就可以了。当然这里有个注意事项：就是我们需要判断这个栈是否为空，如果为空就要返回0 。
• 时间复杂度：O(1)

class MinStack:

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

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

    def pop(self) -> None:
        a = self.stack.pop()
        if self.min_result and a == self.min_result[-1]:
            self.min_result.pop()

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

    def getMin(self) -> int:
        if not self.min_result:
            return 0
        return self.min_result[-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()

• 改进了一下上面的算法，删掉了一些不必要的代码。

class MinStack:

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

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

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

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

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