题目链接:https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof/
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 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 次
解题思路
- main函数就是最小值函数,能通过这个函数找到栈中最小值。
- 如何通过一次操作取出当前栈中最小值
- 在存储数据的栈外,再单独新建一个栈,用于存储最小值
实现算法
- 建两个数据栈,stackA用于存储数据,stackB用于存储最小值
class MinStack {
// stackA 用于存储数据
this.stackA = []
this.countA = 0
// stackB 用于将数据降序存储(栈顶值为最小值)
this.stackB = []
this.countB = 0
}
- 入栈:stackA正常入栈即可,stackB分为两种情况:
- stackB如果没有数据,直接入栈
- 如果传入的item值 <= stackB最小值,入栈
push(item) {
// 如果stackB无数据或者item < stackB栈顶值(最小值)
if (this.countB===0 || this.stackB[this.countB-1]) {
this.stackB[this.countB++] = item
}
// stackA正常入栈
this.stackA[this.countA] = item
}
min() {
return this.stackB[this.countB-1]
}
- 出栈:先进行stackB检测,如果stackA 的栈顶值 === stackB栈顶值,stackB出栈,如果不是则不影响取最小值,stackA正常出栈就好了
pop() {
// 判断stackA栈顶值是否是stackB栈顶值(最小值)
if(this.top() === this.min()) {
// 是,则stackB也出栈,否则只需要stackA出栈即可
delete this.stackB[--this.countB]
}
delete this.stackA[--this.countA]
}
top() {
return this.stackA[this.countA-1]
}
- 额外定义两个方法,top() 获取stackA栈顶值;min()获取stackB栈顶值