Q:
Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.
· push(x) -- Push element x onto stack.
· pop() -- Removes the element on top of the stack.
· top() -- Get the top element.
· getMin() -- Retrieve the minimum element in the stack.
Example:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> Returns -3.
minStack.pop();
minStack.top(); --> Returns 0.
minStack.getMin(); --> Returns -2.
A:
如果是最小值,那么push进来两次,主要目的是在pop掉最小值的时候,能够还剩一个作为“存根”进行比对,这是很巧妙地思路,“存”+比对后,再把第二个pop掉。
测试用例: 6,-2,7,4,-6。
陆续push进去之后,stack里从上至下是:-6,-6,4,7,-2,-2,6,6。
class MinStack {
Stack<Integer> stack = new Stack<>(); //declaration one stack here
int min = Integer.MAX_VALUE; //先声明min, no worry about stack.empty
public void push(int x) {
if (x <= min) {
stack.push(min);
min = x; //单独记录下最小值是什么
}
stack.push(x); //(看上面的解释)
}
public void pop() {
int top = stack.pop(); //是赋值语句,也完成了pop操作
if (top == min){
min = stack.pop(); //之前每次出现min就push进两次,这里pop第二次
}
} //(看下面的代码)
public int top() {
return stack.peek(); //import java.util.Stack
}
public int getMin() {
return min;
}
}
下面这个实现 pop()的版本,含义完全一样,虽然不够简洁,但是更易懂!
(备注里写了如何改动成上面精简版本的步骤和逻辑过程)
public void pop() { if (stack.peek() == min) {
//2.要pop()肯定是针对stack.peek(),那直接stack.peek() = stack.pop(),然后判断是不是等于min
//3. 第一步已经pop掉一次了,这个可以删了stack.pop();
min = stack.peek();
//4. 当初存两个目的就是为了这里,可以retrieve min value
stack.pop();
//5.已经知道min是啥了,可以把old min里的第二个“存根”也pop掉了
}
else { stack.pop();
//1. 上面俩pop(),这里又一个pop(),说明无论如何都得pop()一次
} }