题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
解题思路
要最少的时间复杂度,则用空间换时间。开辟一个新的空间用来存放每次栈中元素的最小值。具体如图所示。
代码示例
# -*- coding:utf-8 -*-
class Solution:
# 定义栈数据结构的对象
def __init__(self):
self.stack = []
self.minValue = [] # 开辟新的空间用来存储每次栈中最小值
# 入栈
def push(self, node):
# write code here
self.stack.append(node)
# 入栈后判断当前栈与当前minValue中栈顶元素的大小,
# 从而确定是否将当前入栈元素存入minValue中,
# 如果minValue中有值且入栈的元素比minValue中栈顶元素小,
# 则说明该元素为此时栈中的最小值,将其存入minValue中
if self.minValue:
while self.minValue[-1] > node:
self.minValue.append(node)
else: # minValue为空的情况,直接存入该元素
self.minValue.append(node)
# 出栈
def pop(self):
# write code here
# 考虑栈为空的情况,返回None
if self.stack == []:
return None
# stack中直接出栈
result = self.stack.pop()
# minValue中如果其栈顶元素与stack出栈的元素相等才出栈
if self.minValue[-1] == result:
self.minValue.pop()
return result
def top(self):
# write code here
if self.stack == []:
return None
else:
return self.stack[-1]
def min(self):
# write code here
if self.minValue == []:
return None
return self.minValue[-1]
if __name__ == '__main__':
s = Solution()
print('----入栈......----')
s.push(7)
s.push(8)
s.push(3)
s.push(2)
s.push(9)
s.push(1)
print('最小值:',s.min())
print('----出栈----')
print(s.pop())
print(s.pop())
print(s.pop())
print(s.pop())
print(s.pop())
print(s.pop())