剑指offer面试题30. 包含min函数的栈(辅助栈)

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

思路

详见链接

代码

class MinStack:
	def __init__(self):
		self.A, self.B = [], []
	def push(self, x:int)->None:
		self.A.append(x)
		if not self.B and self.B[-1] >= x:
			self.B.append(x)
	def pop(self)->None:
		if self.A.pop() == self.B[-1]:
			self.B.pop()
	def top(self)->int:
		return self.A[-1]
	def min(self)->int:
		return self.B[-1]

复杂度

时间复杂度 O(1) ： push(), pop(), top(), min() 四个函数的时间复杂度均为常数级别。
空间复杂度 O(N)： 当共有 N个待入栈元素时，辅助栈 BB 最差情况下存储 N 个元素，使用 O(N) 额外空间。