题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。要求时间复杂度是o(1)。
还是先说一下思路吧,因为每次方元素进栈的时候不能保证栈顶元素都是最小的,所以需要想办法使得栈顶元素始终是最小的元素,排序是一种思路,但是每次排序还设计重新出栈和入栈,想来应该不是这样的。一种思路是可以利用一个辅助栈,相当于是以空间换时间了。具体思路是:当入栈的新元素原先栈顶元素小的话就该元素放入一个辅助栈中,如果新入栈的元素比原先栈顶元素大,则在辅助栈中继续放原先最小的元素。这样一直放下去,当需要取最小元素的时候,就从辅助栈中取栈顶元素即可,这样的话时间复杂度就是o(1)了。而且这么做还有一个好处,就是去了最小元素后,下一次取的时次小的元素(仍然是最小的)。基于这种思路,写出如下代码(已被牛客ac):