题目:
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
实现 MinStack 类:
MinStack() 初始化堆栈对象。
void push(int val) 将元素val推入堆栈。
void pop() 删除堆栈顶部的元素。
int top() 获取堆栈顶部的元素。
int getMin() 获取堆栈中的最小元素
实例:
输入:
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]输出:
[null,null,null,null,-3,null,0,-2]解释:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
思路: 辅助栈
如果一个元素 a 在入栈时,栈里有其它的元素 b, c, d,那么无论这个栈在之后经历了什么操作,只要 a 在栈中,b, c, d 就一定在栈中。那么,我们可以在每个元素 a 入栈时把当前栈的最小值 m 存储起来。在这之后无论何时,如果栈顶元素是 a,我们就可以直接返回存储的最小值 m。
算法:
使用一个辅助栈,与元素栈同步插入与删除,用于存储与每个元素对应的最小值。
1.当一个元素要入栈时,我们取当前辅助栈的栈顶存储的最小值,与当前元素比较得出最小值,将这个最小值插入辅助栈中;
2.当一个元素要出栈时,我们把辅助栈的栈顶元素也一并弹出;
3.在任意一个时刻,栈内元素的最小值就存储在辅助栈的栈顶元素中。
class MinStack {
Stack<Integer> stack;
Stack<Integer> Minstack;
public MinStack() {
stack=new Stack<>();
Minstack=new Stack<>();
}
public void push(int val) {
stack.push(val);
if(!Minstack.isEmpty()){
Minstack.push(Math.min(Minstack.peek(),val));
}else{
Minstack.push(val);
}
}
public void pop() {
stack.pop();
Minstack.pop();
}
public int top() {
return stack.peek();
}
public int getMin() {
return Minstack.peek();
}
}