请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1
示例 1:
输入:
[“MaxQueue”,“push_back”,“push_back”,“max_value”,“pop_front”,“max_value”]
[[],[1],[2],[],[],[]]
输出: [null,null,null,2,1,2]
思路:
这个题目和之前做的取min的思路一模一样。
做法就是维持一个双端队列,保证他是单调递减的,最大值在队首。
然后在每push一个数的时候,我们将这个数和deque中的last数进行对比,如果大于就把deque的last poll掉
在pop的时候,我们需要判断要poll掉的这个数是不是deque的队首,如果是的话就要poll掉。注意 因为Queue和Deque为Integer类型,所以我们在比较的时候需要使用equals函数来进行比较。
class MaxQueue {
//使用双端队列来维护一个递减序列
//注意一下Queue的函数,add/offer 添加元素
//poll 删除队首并返回
//peek 返回队首元素
Queue<Integer> queue;
Deque<Integer> deque;
public MaxQueue() {
queue=new LinkedList<>();
deque=new LinkedList<>();
}
public int max_value() {
return deque.isEmpty()?-1:deque.peekFirst();
}
public void push_back(int value) {
//在push的时候,我们需要把value与deque中的元素进行对比
queue.add(value);
while(!deque.isEmpty()&&deque.peekLast()<value)
//把比value小的都poll掉
deque.pollLast();
//添加到右侧
deque.addLast(value);
}
public int pop_front() {
if(queue.isEmpty())
return -1;
if(queue.peek().equals(deque.peekFirst()))
deque.pollFirst();
return queue.poll();
}
}
/**
* Your MaxQueue object will be instantiated and called as such:
* MaxQueue obj = new MaxQueue();
* int param_1 = obj.max_value();
* obj.push_back(value);
* int param_3 = obj.pop_front();
*/