题目
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
思路
辅助栈+单调栈
- 初始化两个栈:stack1放正常元素,stack2单栈非递增栈保持栈顶是最小值
- push()
- stack1.push(x)
- stack2空或者x<=stack2.peek()
- pop():保持stack1和stack2的一致性
- stack1.pop()弹出栈顶元素
- stack2的栈顶元素等于stack1.pop()弹出的栈顶元素,stack.pop()出栈
- top():stack1不空,返回stack1的栈顶元素,不删除
- min(): stack2不空,返回stack2的栈顶元素
class MinStack {
Deque<Integer> stack1;
Deque<Integer> stack2;
/** initialize your data structure here. */
public MinStack() {
stack1=new LinkedList<>();
stack2=new LinkedList<>();
}
public void push(int x) {
stack1.push(x);
//必须是小于等于
//因为要的就是最小值,保证栈顶永远最小
//x==stack2.peek()防止最小值被在pop()可能会被弹出的可能
if(stack2.isEmpty()||x<=stack2.peek()){
stack2.push(x);
}
}
public void pop() {
if(stack1.pop().equals(stack2.peek())){
stack2.pop();
}
}
public int top() {
if(!stack1.isEmpty()){
return stack1.peek();
}
return 0;
}
public int min() {
if(!stack2.isEmpty()){
return stack2.peek();
}
return 0;
}
}
/**
* 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.min();
*/
时间复杂度O(1)
空间复杂度O(n): n个元素入栈stack1的时候,stack2最差存储n个元素