寻找栈里面最小的元素,python

版权声明:如需转载请注明出处,谢谢 https://blog.csdn.net/qinglv1/article/details/82380452

一.

#这个程序不知道是不是比较严格的正确。不过对于测试的程序来看,还是比较符合的。这个只是涉及到push入栈的早做了,没有涉及到出栈的操作。功能不够完善,但是可以实现找到一个固定栈的最小值

#这个函数的功能是利用时间复杂度为O(1),来实现返回栈里面最小的元素。
#借助一个辅助栈,利用它来存储每次入栈的最小元素。如果只用一个栈,这样的时间复杂度是O(n)
class zhan_min(object):
    stack_list=[]
    min_list=[]
    min_2=None
    def push(self,x):
        if not self.stack_list:
            self.stack_list.append(x)
            self.min_list.append(x)
        else:
            self.stack_list.append(x)
            if x<self.min_list[-1]:
                self.min_list.append(x)
    def min_stack(self):
        if self.min_list:
            self.min_2=self.min_list[-1]
        return self.min_2
if __name__=="__main__":
    s=zhan_min()
    s.push(3)
    s.push(1)
    s.push(2)
    s.push(-2)
    print(s.min_stack())

二.这个是leetcode上面155题的解法,时间复杂度为O(1)

class MinStack(object):
    def __init__(self):
        """
        initialize your data structure here.
        """
        self.stack = []
        self.minStack = []

    def push(self, x):
        """
        :type x: int
        :rtype: nothing
        """
        self.stack.append(x)
        if len(self.minStack) == 0 or x <= self.minStack[-1]:
            self.minStack.append(x)

    def pop(self):
        """
        :rtype: nothing
        """
        if not self.isEmpty():
            if self.top() == self.minStack[-1]:
                self.minStack.pop()
            self.stack.pop()
        return

    def top(self):
        """
        :rtype: int
        """
        if not self.isEmpty():
            return self.stack[-1]

    def getMin(self):
        """
        :rtype: int
        """
        if not self.isEmpty():
            return self.minStack[-1]

    def isEmpty(self):
        return len(self.stack) < 1


if __name__ == "__main__":
    minstack = MinStack()
    minstack.push(2)
    minstack.push(0)
    minstack.push(3)
    minstack.push(0)
    print(minstack.getMin())
    minstack.pop()
    print(minstack.getMin())
    minstack.pop()
    print(minstack.getMin())
    minstack.pop()
    print(minstack.getMin())

猜你喜欢

转载自blog.csdn.net/qinglv1/article/details/82380452