版权声明:如需转载请注明出处,谢谢 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())