设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。 pop() —— 删除栈顶的元素。 top() —— 获取栈顶元素。 getMin() —— 检索栈中的最小元素。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/min-stack 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
最开始有想过使用一个辅助 TreeSet来做,但是考虑到Set 不能保存相同元素,这时如果入栈多个相同的最小值,将出错。 正确思路:利用一个辅助栈用来同步每一次入栈元素时的当前栈的最小值,是一 一对应的关系。 LinkedList 用来实现栈 :addFirst() 、getFirst()、removeFirst() LinkedList 实现了Queue接口:Queue<> queue = new LinkedList<>()、 添加到尾:offer() 取头:peek() 去头:poll()、remove()
class MinStack { private LinkedList<Integer> list1 = new LinkedList<Integer>(); private LinkedList<Integer> list2 = new LinkedList<Integer>(); /** initialize your data structure here. */ public MinStack() { list2.addFirst(Integer.MAX_VALUE); } public void push(int x) { list1.addFirst(x); if(x < list2.getFirst()){ list2.addFirst(x); }else{ list2.addFirst(list2.getFirst()); } } public void pop() { list1.removeFirst(); list2.removeFirst(); } public int top() { return list1.getFirst(); } public int getMin() { return list2.getFirst(); } } /** * 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(); */