Python算法——用O(1)的时间复杂度求栈中最小元素

本文介绍了如何在Python中使用两个栈结构,以O(1)的时间复杂度获取栈中的最小元素。入栈时,若新元素小于辅助栈的最小值,则更新最小值;出栈时,若出栈元素等于辅助栈栈顶元素,则辅助栈也出栈。这种方法确保了始终能在常数时间内获取最小值。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

要求:

       由于栈具有后进先出(Last In First Out, LIFO)的特点,因此 push 和 pop 只需要对栈顶元素进行操作,只能访问到栈顶元素,而无法得到栈中最小的元素。那么,如何求栈中最小元素呢?

解析:

       1.利用一个变量记录栈底的位置,通过遍历栈中的所有元素找出最小值。但是这种方法的时间复杂度为 O(n)。

       2.使用两个栈结构,一个栈用来存储数据,另一个栈用来存储栈的最小元素

       实现思路如下:

       (1)使用 elemStack 和 minStack 两个栈结构,elemStack 用来存储数据,minStack 用来存储 elem 栈的最小元素。

       (2)在入栈时,如果当前入栈的元素比原来栈中的最小值还小,则把这个值压入 minStack 栈中;

       (3)在出栈时,如果当前出栈的元素恰好为 minStack栈中的最小值,则 minStack 栈的栈顶元素也出栈,使得当前最小值变为当前最小值入栈之前的那个最小值。


     假设 elem 栈中元素为3,2,5,1,6那么 min 栈中对应的元素为3,2,1。对应的图示如下:

   

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值