题目
请设计一个栈,除了常规栈支持的pop与push函数以外,还支持min函数,该函数返回栈元素中的最小值。执行push、pop和min操作的时间复杂度必须为O(1)。
解题思路
2个栈实现最小栈
- stack保存栈的数值
- minStack保存最小值
- 注意保持stack和minStack中最小值的同步
代码
import java.util.Deque;
import java.util.LinkedList;
class class MinStack {
Deque<Integer> stack, minStack;
public MinStack() {
stack = new LinkedList<>();
minStack = new LinkedList<>();
}
public void push(int x) {
// 入栈
stack.push(x);
// 如果最小栈为空,x存入最小栈中
// 如果x <= 当前栈中最小值,x存入栈中
if (minStack.isEmpty() || x <= minStack.peek()) minStack.push(x);
}
public void pop() {
// stack直接pop;如果pop出的是当前最小值,最小栈中也弹出当前最小值
if (stack.pop().equals(minStack.peek())) minStack.pop();
}
public int top() {
return stack.peek();
}
public int min() {
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();
*/