设计一个栈-最小栈

设计一个栈

设计一个支持push,pop,top,getMin操作,并且能在常数时间内检索最小元素的栈
push(x)
pop()
top()
getMin()

示例
输入:
[“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“top”,“getMin”]
[[],[-2],[0],[-3],[],[],[],[]]

输出:
[null,null,null,null,-3,null,0,-2]

关于解题

  1. 思路:利用以空间换时间的思想,创建一个辅助栈min_stack(),用于获取stack中的最小值。

  2. 算法流程
    (1)创建两个自定义空栈,一个为原始栈stack(),一个为辅助栈min_stack()
    (2)关于push(),每当有新值入栈时,如果小于等于min_stack()栈顶值即最小 值时,则同时将新值push()进min_stack(),即同时更新栈顶最小值。
    (3)关于pop(),需要判断stack.pop()是否是min_stack()的栈顶元素即最小值,如果是,则同时进行min_stack.pop(),则可以保证min_stack()的栈顶元素是stack()的最小值。
    (4)关于top(),直接返回stack()栈顶元素即可。
    (5)关于getMin(),直接返回min_stack()栈顶元素即可。

  3. 算法复杂度分析
    时间复杂度O(1):入栈,出栈,获取值的所有操作,其时间复杂度都为O(1)。
    空间复杂度O(n):n为总操作数,最坏情况为连续入栈n个数,此时两个栈占用的空间为O(n),即其包含n个元素辅助栈占用线性大小的额外空间。

code

class MinStack:
	def __init__(self):
		self.stack = []
		self.min_stack = []
	def push(self, x: int) -> None:
		self.stack.append(x)
		if not self.min_stack() or x <= self.min_stack[-1]:  # 若辅助栈为空或新入栈元素比辅助栈栈顶值更小
			self.min_stack.append(x)
	def pop(self) -> None:
		if self.stack.pop() == self.min_stack[-1]:
			self.min_stack.pop()
	def top(self) -> int:
		return self.stack[-1]
	def getMin(self) -> int:
		return self.min_stack[-1]

code中部分问题参考
1.关于def __ init __(self):可参考__init __在python中的用法
2.push()代码中,若没有not self.min_stack( )的话,则忽略此时栈stack()为空的情况,即程序访问的self.min_stack[-1]是不存在的,则代码运行后会报越界错误,因此需要加判空条件。
3.pop()返回并删除栈顶元素。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值