定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
思路分析:
要能得到栈内最小元素,容易想到用一个变量来存这个元素,但问题是这个元素出去以后,下一个最小元素呢?解决该问题就得专门用一个数组来存第一小第二小第三小...的元素们了,定义两个数组,主数组arr1用来做栈的出栈和入栈,副数组arr2用来存小元素。重点来了,
入栈时,若该元素小于等于当前最小元素,那该元素也要往arr2中放;
出栈时,若该元素是当前最小元素,arr2尾部的该元素也要一同出来;
由此,arr2中存的元素越往尾部越小,每次读尾部最后一个元素就是最小元素。
代码实现:
var MinStack = function() {
this.arr1 = []
this.arr2 = []
};
/**
* @param {number} x
* @return {void}
*/
MinStack.prototype.push = function(x) {
this.arr1.push(x)
if (!this.arr2.length || (this.arr2.length && x <= this.arr2[this.arr2.length - 1])) {
this.arr2.push(x)
}
};
/**
* @return {void}
*/
MinStack.prototype.pop = function() {
const e = this.arr1.pop()
if (this.arr2[this.arr2.length - 1] === e) {
this.arr2.pop()
}
return e
};
/**
* @return {number}
*/
MinStack.prototype.top = function() {
return this.arr1[this.arr1.length - 1]
};
/**
* @return {number}
*/
MinStack.prototype.min = function() {
return this.arr2[this.arr2.length - 1]
};