一、题目描述
实现一个栈,该栈带有出栈(pop)、入栈(push)、取最小元素(get_min)三个方法,要保证这三个方法的时间复杂度都是O(1)。
二、解题思路
可能大部分在看到这个题后,第一反应是创建一个整型变量min,用来存储栈中的最小元素,每新进来一个元素,就让它和最小元素比较,只要它比最小元素小,就更新它为最小元素。当调用get_min时,直接返回min的值即可。
以上算法错误之处在于只考虑了进栈的情况,当栈顶元素出栈时,此时栈中的最小值我们并不能知道是多少。
所以,只存一个最小值是不够的,我们需要把栈中的历史最小值都存储下来。
详细步骤如下:
1、设原来的栈为A,我们新建的存储历史最小值的栈称之为B。
2、当第一个元素进入A时,此时A的最小值就是该元素,所以把该元素也放入栈B中。
3、每当新进来一个元素,都要比较新元素和栈A当前最小值的大小,如果新元素小于栈A当前最小值,则让新元素也进入栈B中。以此办法让栈B的栈顶元素始终是栈A的最小值。
4、当栈A有元素出栈时,如果出栈元素是栈A当前最小值,此时让栈B的栈顶元素也出栈。
5、当调用get_min时,返回栈B的栈顶元素,即栈A的最小值。
该方法实现了出栈(pop)、入栈(push)、取最小元素(get_min)的时间复杂度都是O(1),但是空间复杂度为O(n)
三、代码实现
class MinStack:
def __init__(self):
self.main_stack=[]
self.min_stack