设计一个支持push,pop,top等操作并且可以在O(1)时间内检索出最小元素的堆栈。
- push(x)–将元素x插入栈中
- pop()–移除栈顶元素
- top()–得到栈顶元素
- getMin()–得到栈中最小元素
数据范围:
操作命令总数 [0,100]。
样例:
MinStack minStack = new MinStack();
minStack.push(-1);
minStack.push(3);
minStack.push(-4);
minStack.getMin(); --> Returns -4.
minStack.pop();
minStack.top(); --> Returns 3.
minStack.getMin(); --> Returns -1.
代码:
class MinStack {
//主栈,存储所有元素
private Deque<Integer>stack;
//辅助栈,存储当前最小值
private Deque<Integer>minStack;
/** initialize your data structure here. */
public MinStack() {
//初始化主栈和辅助栈
stack = new LinkedList<>();
minStack = new LinkedList<>();
//辅助栈初始时存储整型最大值
minStack.push(Integer.MAX_VALUE);
}
public void push(int x) {
//元素压入主栈
stack.push(x);
//新值小于辅助栈当前最小值时,压入新值
//否则压入辅助栈当前最小值
minStack.push(Math.min(minStack.peek(),x));
}
public void pop() {
//弹出主栈元素
stack.pop();
//弹出辅助栈元素
minStack.pop();
}
public int top() {
//返回主栈栈顶元素
return stack.peek();
}
public int getMin() {
//返回辅助栈栈顶元素(当前最小值)
return minStack.peek();
}
}
/**
* 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();
*/