剑指 Offer 30.包含min函数的栈
题目描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 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.
提示:
- 各函数的调用总次数不超过 20000 次
思路
- 如果仅仅是一个栈来得到最小元素的话必须就需要遍历这个栈,那么时间复杂度就会是O(n),因此我们需要一个
辅助栈来存储数据栈中的最小元素
- 一个数据栈用于实现基本的栈的入栈,出栈等基本操作,一个辅助栈用于存储数据栈的最小元素
- 使用辅助栈来存储最小值,每次
push()
时,如果新元素比当前最小值小,则将新元素入辅助栈;每次pop()
时,如果弹出的元素是当前最小值,则将辅助栈栈顶元素弹出。top()
方法直接返回栈顶元素,min()
方法返回辅助栈的栈顶元素即可。
代码
/**
* initialize your data structure here.
*/
var MinStack = function() {
//数据栈 保证完整的栈的基本操作
this.Stack = []
//辅助栈 记录数据栈中最小值
this.MinStack = [Infinity]
//要保证两栈的一致性
};
/**
* @param {number} x
* @return {void}
*/
MinStack.prototype.push = function(x) {
//数据栈正常压栈
this.Stack.push(x)
//辅助栈判断 辅助栈为空直接压栈 不为空 与辅助栈栈顶元素判断大小 如果小或者相等压入
if(this.MinStack.length == 0 || x <= this.MinStack[this.MinStack.length-1]){
this.MinStack.push(x)
}
};
/**
* @return {void}
*/
MinStack.prototype.pop = function() {
//出栈
//数据栈正常出栈
//如果辅助栈栈顶元素和出栈元素相等 辅助栈栈顶元素也出栈
var number = this.Stack.pop()
if(number == 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()
*/