Tips:本文字数1621,阅读大约10分钟左右。分三个板块,问题描述,问题分析,解决思路(代码)。
问题描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
问题分析:
其实要实现栈,在python当中我们很容易想到的数据形式就是列表,列表自带的函数和方法如下图所示,可以很轻松的实现。
但是该题中很重要的一句是**调用 min、push 及 pop 的时间复杂度都是 O(1)。**普通的push和pop时间复杂度都是O(1),意为直接压入或弹出某元素即可。可是python自带的min()函数需要遍历整个列表以后才能返回最小值,那么列表中的n个数就构成了O(n)的复杂度,与题意不符。
解决思路:
主栈设为A,在建立一个辅助栈B,在我们正常使用主栈A进行压入或弹出元素操作时,用辅助栈B来记录一个降序的元素列表。
具体用图来解释就是说,假如我们现在开始向主栈A中压入元素
第一步:A压入9,此时