面试之手撕代码二
程序员文章站
2024-03-21 22:05:28
...
(二)
设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。
解法:关键是如何在常数时间做到,解法为双栈(空间换时间),一个存数据,一个存最小值,push的同时,也在第二个栈push最小值(或者push的时候只比较当前值与最小栈的栈顶,不更大push,pop的同时判断是否与第二个栈顶相同,同则pop)。
Python版(第一种解法):
class MinStack(object):
def __init__(self):
"""
initialize your data structure here.
"""
self.result = []
self.min = []
def push(self, x):
"""
:type x: int
:rtype: None
"""
self.result.append(x)
if len(self.min) == 0:
self.min.append(x)
else:
if x < self.min[-1]:
self.min.append(x)
else:
self.min.append(self.min[-1])
def pop(self):
"""
:rtype: None
"""
self.min.pop()
return self.result.pop()
def top(self):
"""
:rtype: int
"""
return self.result[-1]
def getMin(self):
"""
:rtype: int
"""
if len(self.min) == 0:
return []
return self.min[-1]
上一篇: ASP生成HTML asp