题目
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
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.
注意:pop、top 和 getMin 操作总是在 非空栈 上调用。
方法:使用两个栈完成,一个普通栈normal用来存数据,还有一个最小数栈min,来存最小的数。
push操作:
将数据newNum压入normal栈然后判断min栈中是否为空:
- 如果为空,则将newNum也压入min栈
- 如果不为空,则比较newNum和min栈的栈顶元素哪一个更小
- 如果newNum更小或者两者相等,则newNum也压入min栈中
- 如果min栈中栈顶元素小,则min栈不压入任何内容
pop操作:
弹出normal栈中的栈顶元素记为value,然后将该元素与min栈的栈顶元素进行比较,如果value等于min栈的栈顶元素,将min栈的栈顶元素也弹出,如果value大于min栈的栈顶元素时,min栈不弹出栈顶元素,只有normal栈弹出栈顶元素。
getMin操作:
min栈的栈顶元素即是最小元素
top操作:
top就是获得栈顶元素
代码
class MinStack {
private Stack<Integer> normal = new Stack<>();
private Stack<Integer> min = new Stack<>();
/** initialize your data structure here. */
public MinStack() {
}
public void push(int val) {
normal.push(val);
if (min.isEmpty() || val <= min.peek()){
min.push(val);
}
}
public void pop() {
if ((int)normal.pop() == min.peek() ){//两个Integer不能这么比较,默认不会拆箱
min.pop();
}
}
public int top() {
return normal.peek();
}
public int getMin() {
return min.peek();
}
}