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
思路:用一个temp栈保存当前最小值,当要push一个值进栈时,与temp栈顶元素比较,若小于等于temp栈顶元素,则将这个值push到两个栈中。
pop一个数时,与temp栈顶元素比较,若等于temp栈顶元素,则pop temp栈顶元素。
代码:
class MinStack {
Stack<Integer> stack = new Stack<Integer> ();
Stack<Integer> temp = new Stack<Integer> ();
public void push(int x) {
if(stack.isEmpty()||x<=temp.peek()){
temp.push(x);
}
stack.push(x);
}
public void pop() {
int x = stack.peek(); //此处是个坑,stack.peek()==temp.peek() 永远是false
if(x==temp.peek()){
temp.pop();
}
stack.pop();
}
public int top() {
return stack.peek();
}
public int getMin() {
return temp.peek() ;
}
}
Stack<Integer> stack = new Stack<Integer> ();
Stack<Integer> temp = new Stack<Integer> ();
public void push(int x) {
if(stack.isEmpty()||x<=temp.peek()){
temp.push(x);
}
stack.push(x);
}
public void pop() {
int x = stack.peek(); //此处是个坑,stack.peek()==temp.peek() 永远是false
if(x==temp.peek()){
temp.pop();
}
stack.pop();
}
public int top() {
return stack.peek();
}
public int getMin() {
return temp.peek() ;
}
}