
此题中的push、pop、top操作都是Stack及Deque等类可以支持的,它们的时间复杂度为O(1)。难点在于getMain方法,获取堆栈中的最小元素,如果遍历整个堆栈区找寻最小值时间复杂度为O(n),想要将时间复杂度优化到O(1)就需要辅助栈。
push操作:

辅助栈跟原始栈的操作是一致的,但是辅助栈的数据需要取push值跟当前辅助栈值的最小值,这样在获取栈内最小值时直接从辅助栈进行peek获取栈顶数据即可。
辅助栈的步骤跟原始栈的步骤一致,push跟pop都同步入栈跟出栈,要达到两个栈操作的一致。pop操作就简单了,不需要判断直接两个栈进行pop即可,就不画图了,看下面代码。
辅助栈初始化时需要写入一个最大值,否则第一次比对栈顶没有内容会报错。class MinStack {
ArrayDeque<Integer> stack;
ArrayDeque<Integer> minStack;
public MinStack() {
stack = new ArrayDeque<>();
minStack = new ArrayDeque<>();
minStack.push(Integer.MAX_VALUE);
}
public void push(int val) {
stack.push(val);
minStack.push(Math.min(minStack.peek(), val));
}
public void pop() {
stack.pop();
minStack.pop();
}
public int top() {
return stack.peek();
}
public int getMin() {
return minStack.peek();
}
}