设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
- 方法:
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。 - 示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
精选方案
思路
最小栈的用法是以空间换取时间的思想,在创建栈的同时创建一个辅助的最小栈
算法
设置两个类的字段分别是栈和最小栈。
push:最小栈用来存放每次栈push之后的与最小栈顶比较的较小值。
pop:当栈pop操作后,pop的值如果是最小值,则将最小栈的栈顶也pop。这样最小栈接下来的栈顶依然是以push为顺序的最小栈元素。
getMin:返回最小栈的栈顶元素
代码
```class MinStack {
/** initialize your data structure here. */
//使用一个额外的辅助栈,push时,当x小于等于辅助栈的栈顶时,push辅助栈。pop时,当辅助栈的栈顶和当前栈相同时在一起POP。
//这样辅助栈的栈顶会一直保持最小。
Stack<Integer> min;
Stack<Integer> stack;
public MinStack() {
stack = new Stack<Integer>();
min = new Stack<Integer>();
}
public void push(int x) {
stack.push(x);
if(min.empty()){
min.push(x);
}else{
if(x<=min.peek()){
min.push(x);
}
}
}
public void pop() {
if(min.peek().equals(stack.peek())){
min.pop();
}
if(!stack.empty()){
stack.pop();
}else
throw new RuntimeException("11");
}
public int top() {
if(!stack.empty()){
return stack.peek();
}else
throw new RuntimeException("11");
}
public int getMin() {
if(!min.empty()){
return min.peek();
}else{
throw new RuntimeException("11");
}
}
}