定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 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.
解题思路
使用一个辅助栈来存储前i个栈中元素的最小值。
class MinStack {
LinkedList<Integer> stack;
LinkedList<Integer> minimumNum;
public MinStack() {
stack = new LinkedList<>();
minimumNum = new LinkedList<>();
}
public void push(int x) {
stack.add(x);
if (minimumNum.isEmpty()) minimumNum.add(x);
else
minimumNum.add(
x < minimumNum.get(minimumNum.size() - 1) ? x : minimumNum.get(minimumNum.size() - 1));
}
public void pop() {
if (stack.isEmpty()) return;
stack.remove(stack.size() - 1);
minimumNum.remove(minimumNum.size() - 1);
}
public int top() {
if (stack.isEmpty()) return -1;
return stack.get(stack.size() - 1);
}
public int min() {
if (minimumNum.isEmpty()) return -1;
return minimumNum.get(minimumNum.size() - 1);
}
}