题目:
1670. 设计前中后队列
请你设计一个队列,支持在前,中,后三个位置的 push
和 pop
操作。
请你完成 FrontMiddleBack
类:
FrontMiddleBack()
初始化队列。void pushFront(int val)
将val
添加到队列的 最前面 。void pushMiddle(int val)
将val
添加到队列的 正中间 。void pushBack(int val)
将val
添加到队里的 最后面 。int popFront()
将 最前面 的元素从队列中删除并返回值,如果删除之前队列为空,那么返回-1
。int popMiddle()
将 正中间 的元素从队列中删除并返回值,如果删除之前队列为空,那么返回-1
。int popBack()
将 最后面 的元素从队列中删除并返回值,如果删除之前队列为空,那么返回-1
。
请注意当有 两个 中间位置的时候,选择靠前面的位置进行操作。比方说:
- 将
6
添加到[1, 2, 3, 4, 5]
的中间位置,结果数组为[1, 2, 6, 3, 4, 5]
。 - 从
[1, 2, 3, 4, 5, 6]
的中间位置弹出元素,返回3
,数组变为[1, 2, 4, 5, 6]
。
示例 1:
输入: ["FrontMiddleBackQueue", "pushFront", "pushBack", "pushMiddle", "pushMiddle", "popFront", "popMiddle", "popMiddle", "popBack", "popFront"] [[], [1], [2], [3], [4], [], [], [], [], []] 输出: [null, null, null, null, null, 1, 3, 4, 2, -1]解释: FrontMiddleBackQueue q = new FrontMiddleBackQueue(); q.pushFront(1); // [1] q.pushBack(2); // [1, 2] q.pushMiddle(3); // [1, 3, 2] q.pushMiddle(4); // [1, 4, 3, 2] q.popFront(); // 返回 1 -> [4, 3, 2] q.popMiddle(); // 返回 3 -> [4, 2] q.popMiddle(); // 返回 4 -> [2] q.popBack(); // 返回 2 -> [] q.popFront(); // 返回 -1 -> [] (队列为空)
提示:
1 <= val <= 109
- 最多调用
1000
次pushFront
,pushMiddle
,pushBack
,popFront
,popMiddle
和popBack
。
解答:
代码:
class FrontMiddleBackQueue {private Deque<Integer> left;private Deque<Integer> right;public FrontMiddleBackQueue() {left=new ArrayDeque<>();right=new ArrayDeque<>();}//调整长度,保证0<=right.size()-left.size()<=1//从而保证可以在正中间插入删除元素private void balance(){if(left.size() > right.size()){right.addFirst(left.pollLast());}else if(right.size()>left.size()+1){left.addLast(right.pollFirst());}}public void pushFront(int val) {left.addFirst(val);balance();}public void pushMiddle(int val) {if(left.size()<right.size()){left.addLast(val);}else{right.addFirst(val);}}public void pushBack(int val) {right.addLast(val);balance();}public int popFront() {if(right.isEmpty()){//整个队列为空return -1;}int val=left.isEmpty()?right.pollFirst():left.pollFirst();balance();return val;}public int popMiddle() {if(right.isEmpty()){//整个队列为空return -1;}if(left.size()==right.size()){return left.pollLast();}return right.pollFirst();}public int popBack() {if(right.isEmpty()){return -1;}int val=right.pollLast();balance();return val;}
}/*** Your FrontMiddleBackQueue object will be instantiated and called as such:* FrontMiddleBackQueue obj = new FrontMiddleBackQueue();* obj.pushFront(val);* obj.pushMiddle(val);* obj.pushBack(val);* int param_4 = obj.popFront();* int param_5 = obj.popMiddle();* int param_6 = obj.popBack();*/
结果: