栈
实现一个栈,该栈带有出栈(pop)、入栈(push)、取最小元素(getMin)3
个方法。要保证这3个方法的时间复杂度都是O(1)。
定义一个main变量,每进来一个最小的元素就进行赋值,在java中使用getMin方法过滤每个进栈的元素,找到最小的那个,每当进来一个新的元素都会和已经存在的值进行比较,如果小于当前栈内的最小值则对main变量进行赋值,如果大于则不做任何改变。
那么这种做法的话呢,显然不太好,因为只考虑了进栈场景,没有考虑出栈场景。
假设已经进展的元素有4,9,7,3;那么在第一次出栈的时候当前最小值出栈了,栈内已经没有之前最小值得记录了,这时的最小值变成了4,显然不是我们想要的结果。
接下来优化一下:
- 设原有的栈叫作栈A,此时创建一个额外的“备胎”栈B,用于辅助栈A。
- 当第1个元素进入栈A时,让新元素也进入栈B。这个唯一的元素是栈A的当
前最小值。 - 之后,每当新元素进入栈A时,比较新元素和栈A当前最小值的大小,如果小
于栈A当前最小值,则让新元素进入栈B,此时栈B的栈顶元素就是栈A当前最小值。 - 每当栈A有元素出栈时,如果出栈元素是栈A当前最小值,则让栈B的栈顶元
素也出栈。此时栈B余下的栈顶元素所指向的,是栈A当中原本第2小的元素,代替刚
才的出栈元素成为栈A的当前最小值。(备胎转正。) - 当调用getMin方法时,返回栈B的栈顶所存储的值&#