一、实现一个特殊功能的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作。
输入
[[1,3],[1,2],[1,1],[3],[2],[3]]
返回值
[1,2]
备注:有三种操作种类,op1表示push,op2表示pop,op3表示getMin。你需要返回和op3出现次数一样多的数组,表示每次getMin的答案。
思想:2个栈实现,一个进行基本功能操作,另一个记录最小元素
def getMinStack(self, op):
stack, minstack = [], [] #开辟一个栈记录最小值,最小值放栈顶
res = []
for i in range(len(op)):
if op[i][0] == 1: #1表示要做push操作,如[1,3] 对3push
stack.append(op[i][1])
if not minstack or op[i][1] < minstack[-1]:
#入栈时的元素如果小于最小栈的栈顶元素,需入minstack
minstack.append(op[i][1])
if op[i][0] == 2:
if minstack and stack[-1] == minstack[-1]:
minstack.pop()
stack.pop()
if op[i][0] == 3:
res.append(minstack[-1])
return res