题目描述:
设计一个栈结构,满足一下条件:min,push,pop操作的时间复杂度为O(1)。
欢迎大家转发,一起传播知识和正能量,帮助到更多人。辛苦大家转发时注明出处(也是咱们公益编程交流群的入口网址),刘经纬老师共享知识相关文件下载地址为:http://liujingwei.cn
解题思路:
普通栈的pop,push操作的时间复杂度为O(1),但通过遍历的方法找到最小元素的时间复杂度为O(n)。我们采用空间换取时间的办法来讲第时间复杂度,使用两个栈结构,一个栈用来存储数据,另一个栈用来存储最小元素。
具体步骤:
步骤1:使用 elem 和 min 两个栈结构,elem 用来存储数据,min 用来存储 elem 栈的最小元素。
步骤2:在入栈时,如果当前入栈的元素比原来栈中的最小值还小,则把这个值压入 min 栈中;
步骤3:在出栈时,如果当前出栈的元素巧好为当前栈中的最小值,则 min 栈的栈顶元素也出栈,是的当前最小值变为其入栈之前的那个最小值。
Python实现:
如图1,定义栈结构。
如图2,图3,实例化栈结构,进行出栈入栈操作。