题目
设计一个支持 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()添加元素、pop()获得栈顶元素并删除、peek()获取栈顶元素但不删除。性质跟栈的性质一样,直接用就行
- 题目中多了一个getMin(),获取最小值。每次push()的时候选取最小值,对最小值进行赋值,就可以得到最小值
- 可是当每次pop()删除的时候,把栈中的最小值删除了,我们的最小值还没有变化就会出错
- 在最小值的下面多push() 一次 次小值 ,在pop()的时候对最小值进行赋值,并删除之前多压入的次小值
java
class MinStack {
private Stack<Integer> stack;
private int minVal = Integer.MAX_VALUE;
/** initialize your data structure here. */
public MinStack() {
stack = new Stack<>();
}
public void push(int x) {
if(minVal >= x) {
stack.push(minVal); // 每一个最小值下面压一个之前的最小值(次小值),方便赋值
minVal = x;
}
stack.push(x);
}
public void pop() {
// 当栈顶元素为最小值时,它的下面是次小值,将次小值设置为最小值。将两个值一起出栈
if(stack.pop() == minVal) {
minVal = stack.pop();
}
}
public int top() {
return stack.peek(); // 获取栈顶元素,单不删除栈顶元素
}
public int getMin() {
return minVal;
}
}
/**
* Your MinStack object will be instantiated and called as such:
* MinStack obj = new MinStack();
* obj.push(x);
* obj.pop();
* int param_3 = obj.top();
* int param_4 = obj.getMin();
*/