设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。 pop() —— 删除栈顶的元素。 top() —— 获取栈顶元素。 getMin() —— 检索栈中的最小元素。
示例:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 输入: ["MinStack","push","push","push","getMin","pop","top","getMin"] [[],[-2],[0],[-3],[],[],[],[]] 输出: [null,null,null,null,-3,null,0,-2] 解释: MinStack minStack = new MinStack(); minStack.push(-2); minStack.push(0); minStack.push(-3); minStack.getMin(); --> 返回 -3. minStack.pop(); minStack.top(); --> 返回 0. minStack.getMin(); --> 返回 -2.
提示:
pop、top 和 getMin 操作总是在 非空栈 上调用。
来源:力扣(LeetCode) 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解: 本题可能一看到题目,有可能会想到利用PriorityQueue来解决问题,但是PriorityQueue无法同时满足返回最小和栈结构。本题可以创建一个辅助栈专门存放当前最小的元素,返回栈中最小的元素直接操作最小栈中的元素即可,其他的操作按照正常的栈来操作。用不同的集合类可能会造成时间复杂度上的差异,这里使用Deque。
具体代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 class MinStack { Deque<Integer> stack; Deque<Integer> minStack; public MinStack () { stack = new ArrayDeque(); minStack = new ArrayDeque(); minStack.push(Integer.MAX_VALUE); } public void push (int x) { stack.push(x); minStack.push(Math.min(minStack.peek(), x)); } public void pop () { stack.pop(); minStack.pop(); } public int top () { return stack.peek(); } public int getMin () { return minStack.peek(); } }