1、题目描述:
实现一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作。
【要求】
1.pop、push、getMin操作的时间复杂度都是O(1)。
2.设计的栈类型可以使用现成的栈结构。
2、题解:
方法1:
思路1:
设置两个栈,stackData保存正常的数据,辅助栈stackMin保存正常栈中的最小值,也就是当前栈中的最小值
才能进入stackMin(stackMin长度会<=正常栈的长度)
push():
两个栈都不用判断边界,stackDate无论什么情况下,都可以添加元素;对于stackMin辅助栈,如果为空,直接入栈
否则,就判断新入栈的值与原栈的最小值的大小,小的才能进入辅助栈
pop():
stackData判断边界,然后如果出栈正好是当前栈里的最小值,才让stackMin也出栈,返回staclData;里的出栈元素
getMin():
stackMin判断边界,返回最后一个值
代码:
class getMinStack1:
def __init__(self):
self.stackData = []
self.stackMin = []
def push(self,num):
if not self.stackMin:
self.stackMin.append(num)
elif num <= self.getMin():
self.stackMin.append(num)
self.stackData.append(num)
def pop(self):
if not self.stackData:
return "Your stack is empty."
value = self.stackData.pop()
if value == self.getMin():
self.stackMin.pop()
return value
def getMin(self):
if not self.stackMin:
return "Your stack is empty."
return self.stackMin[-1]
方法2:
思路2:
设置两个栈,stackData保存正常的数据,辅助栈stackMin保存正常栈中的最小值,新入栈stackData中的元素,
辅助栈stackMin也会进入最小值。(stackMin长度与正常栈的长度相等)
push():
两个栈都不用判断边界,stackDate无论什么情况下,都可以添加元素;对于stackMin辅助栈,如果为空,直接入栈
否则,小与当前栈的最小值就直接入栈,大于当前栈的最小值,就让stackMin[-1]入栈
pop():
stackData判断边界,让stackMin出栈,返回staclData;里的出栈元素
getMin():
stackMin判断边界,返回最后一个值
代码:
class getMinStack2:
def __init__(self):
self.stackData = []
self.stackMin = []
def push(self,num):
if not self.stackMin:
self.stackMin.append(num)
elif num <= self.getMin():
self.stackMin.append(num)
else:
self.stackMin.append(self.stackMin[-1])
self.stackData.append(num)
def pop(self):
if not self.stackData:
return "Your stack is empty."
self.stackMin.pop()
return self.stackData.pop()
def getMin(self):
if not self.stackMin:
return "Your stack is empty."
return self.stackMin[-1]