设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() ——检索栈中的最小元素。来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/min-stack
示例:
输入:
[“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.
提示:
pop、top 和 getMin 操作总是在 非空栈 上调用。
题目分析:
本题要求常数时间内,所以不能让所有元素进栈后再遍历,可以用一个辅助栈来解决。创建两个栈st和minst,st用来使所有元素入栈,minst中使当前最小值元素入栈。每个元素压入st时,与minst中栈顶元素作比较,得到当前最小元素,将最小元素压入minst,如果最小元素值不变就不压入minst。当要弹出栈中元素时,首先弹出st栈中的栈顶元素,如果栈顶元素是当前最小值元素,也将minst栈顶元素弹出。如果要获取当前栈栈顶元素,根据题意获取的是st的栈顶元素,返回st的栈顶元素即可。要获取当前栈中最小值,返回minst中的栈顶元素即可。
因为pop、top 和 getMin 操作总是在 非空栈 上调用,所以不用判断栈是否为空。
要注意的是,将元素压入st中时,与minst中栈顶元素作比较条件是小于等于,不是小于,参考全部压入相同元素的例子,如果相同时不压入,弹出栈顶元素时就会出错。
class MinStack {
Stack<Integer> st;
Stack<Integer> minst;
public MinStack() {
st=new Stack<>();
minst=new Stack<>();
}
public void push(int x) {
st.push(x);
if(minst.empty() || x<=minst.peek()) minst.push(x);//如果minst为空,说明x为第一个元素,即为当前最小值。
//注意判断条件是小于等于
}
public void pop() {
int t=st.peek();
st.pop();
if(t==minst.peek()) minst.pop();
}
public int top() {
return st.peek();
}
public int getMin() {
return minst.peek();
}
}