题目描述:
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
算法思想:
要想在常数时间内检索到最小元素,我们可以设置一个辅助栈,栈顶记录了当前栈中的最小值,辅助栈与主栈同进出,这样的话,就保证了对栈顶元素即最小值的更新
代码如下:
class MinStack {
Deque<Integer> xStack;
Deque<Integer> minStack;
/** initialize your data structure here. */
public MinStack() {
xStack = new LinkedList();
minStack = new LinkedList();
minStack.push(Integer.MAX_VALUE);
}
public void push(int x) {
xStack.push(x);
minStack.push(Math.min(minStack.peek(), x));
}
public void pop() {
xStack.pop();
minStack.pop();
}
public int top() {
return xStack.peek();
}
public int getMin() {
return minStack.peek();
}
}
复杂度分析:
时间复杂度:所有操作,时间复杂度均为O(1)
空间复杂度:利用了相同大小的辅助栈,则O(n)
对于代码的小注释:
(1)日常开发中,保存一组数组,最常用的是ArrayList 和 LinkedList,前者是基于数组实现,后者是基于双向链表实现,并且后者同时实现了List接口和Deque接口,Deque接口是双端队列,既能当作队列使用,也能当作栈使用,现在都队列和栈都使用ArrayList 和 LinkedList。
(2)栈中peek()方法和pop()方法的区别:peek()方法是查看当前栈顶元素,并不会出栈。