设计一个有getmin功能的栈

【题目】

  实现一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作。

【要求】

  1、 pop、push、getMin操作的时间复杂度都是O(1)
  2.、设计的栈类型可以使用现成的栈结构

#python3.5
class NewStack1:
def __init__(self):
self.stackData = []
self.stackMin = []

def push(self, newNum):
self.stackData.append(newNum)
if len(self.stackMin) == 0 or newNum <= self.getMin():
self.stackMin.append(newNum)

def pop(self):
if len(self.stackData) == 0:
raise Exception("stack is empty!")
value = self.stackData.pop()
if self.getMin() == value:
self.stackMin.pop()
return value

def getMin(self):
if len(self.stackMin) == 0:
raise Exception("stack is empty!")
return self.stackMin[-1]