思路
本题难点是将 min() 函数复杂度降为 O(1),可通过建立辅助栈实现。
数据栈 A : 栈 A 用于存储所有元素,保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常逻辑。
辅助栈 B : 栈 B中存储栈 A中所有 非严格降序 的元素,则栈 A中的最小元素始终对应栈 B的栈顶元素,即 min() 函数只需返回栈 B的栈顶元素即可。
代码
/**
* initialize your data structure here.
*/
var MinStack = function() {
this.stack = [];
this.minstack = [];//辅助栈,用来存储栈的非严格降序的元素
};
/**
* @param {number} x
* @return {void}
*/
MinStack.prototype.push = function(x) {
this.stack.push(x);
if(!this.minstack.length){
this.minstack.push(x);
}else {
if(x <= this.minstack[this.minstack.length-1]) this.minstack.push(x);
}
};
/**
* @return {void}
*/
MinStack.prototype.pop = function() {
var x = this.stack.pop();
if(x==this.minstack[this.minstack.length-1]) this.minstack.pop();
};
/**
* @return {number}
*/
MinStack.prototype.top = function() {
return this.stack[this.stack.length-1];
};
/**
* @return {number}
*/
MinStack.prototype.min = function() {
return this.minstack[this.minstack.length-1];
};
/**
* Your MinStack object will be instantiated and called as such:
* var obj = new MinStack()
* obj.push(x)
* obj.pop()
* var param_3 = obj.top()
* var param_4 = obj.min()
*/