设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-stack
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
import java.util.Stack;
public class MinStack {
private Stack<Integer> dataStack;
private Stack<Integer> minStack;
public MinStack(){
dataStack = new Stack<>();
minStack = new Stack<>();
}
public void push(int x){
dataStack.push(x);
if(minStack.isEmpty() || x <= minStack.peek()){
minStack.push(x);
}
}
public void pop(){
int x = dataStack.pop();
if( x == minStack.peek()){
minStack.pop();
}
}
public int top(){
return dataStack.peek();
}
public int getMin(){
return minStack.peek();
}
public static void main(String[] args) {
MinStack dataStack = new MinStack();
dataStack.push(-3);
dataStack.push(0);
dataStack.push(-2);
System.out.println(dataStack.top());
dataStack.pop();
System.out.println(dataStack.top());
System.out.println(dataStack.getMin());
}
}