最小栈
题目描述
设计一个支持 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();
*/