定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的 min 函数(时间复杂度应为 $O(1)$)。
Solutions
首先要理解题意,这里用辅助栈的形式存储最小的结果。
# -*- coding:utf-8 -*-
class Solution:
def __init__(self):
self.data = []
self.mins = []
def push(self, node):
# write code here
self.data.append(node)
if len(self.mins) > 0 and self.mins[-1] < node:
self.mins.append(self.mins[-1])
else:
self.mins.append(node)
def pop(self):
# write code here
assert len(self.data) > 0 and len(self.mins) > 0
self.data.pop()
self.mins.pop()
def top(self):
# write code here
return self.data[-1]
def min(self):
# write code here
assert len(self.data) > 0 and len(self.mins) > 0
return self.mins[-1]