生活随笔
收集整理的這篇文章主要介紹了
LeetCode 1670. 设计前中后队列(deque)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
文章目錄
1. 題目
請你設計一個隊列,支持在前,中,后三個位置的 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);
q
.pushBack(2);
q
.pushMiddle(3);
q
.pushMiddle(4);
q
.popFront();
q
.popMiddle();
q
.popMiddle();
q
.popBack();
q
.popFront(); 提示:
1 <= val
<= 10^9
最多調用
1000 次 pushFront, pushMiddle, pushBack, popFront, popMiddle 和 popBack 。
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/design-front-middle-back-queue
著作權歸領扣網絡所有。商業轉載請聯系官方授權,非商業轉載請注明出處。
2. 解題
- 使用1個雙端隊列,插入 Mid 的時間復雜度為O(n)
class FrontMiddleBackQueue {deque
<int> q
;int n
= 0;
public:FrontMiddleBackQueue() {}void pushFront(int val
) {q
.push_front(val
);n
++;}void pushMiddle(int val
) {int mid
= (n
)/2;q
.insert(q
.begin()+mid
, val
);n
++;}void pushBack(int val
) {q
.push_back(val
);n
++;}int popFront() {if(q
.empty())return -1;int ans
= q
.front();q
.pop_front();n
--;return ans
;}int popMiddle() {if(q
.empty())return -1;int mid
= (n
-1)/2;int ans
= q
[mid
];q
.erase(q
.begin()+mid
);n
--;return ans
;}int popBack() {if(q
.empty())return -1;int ans
= q
.back();q
.pop_back();n
--;return ans
;}
};
96 ms 20.2 MB C++
- 使用 2 個雙端隊列,永遠使得 q1.size = q2.size or q1.size = q2.size + 1,這樣可以在 q1 的隊尾取得 Mid,時間復雜度O(1)
class FrontMiddleBackQueue {deque
<int> q1
, q2
;int v
, ans
;
public:FrontMiddleBackQueue() {}bool sameLen(){return q1
.size() == q2
.size();}void pushFront(int val
) {if(q1
.empty()){q1
.push_front(val
);return;}if(sameLen())q1
.push_front(val
);else{q1
.push_front(val
);v
= q1
.back();q1
.pop_back();q2
.push_front(v
);}}void pushMiddle(int val
) {if(q1
.empty()){q1
.push_front(val
);return;}if(sameLen()){q1
.push_back(val
);}else{v
= q1
.back();q1
.pop_back();q2
.push_front(v
);q1
.push_back(val
);}}void pushBack(int val
) {if(q1
.empty()){q1
.push_front(val
);return;}if(sameLen()){q2
.push_back(val
);v
= q2
.front();q2
.pop_front();q1
.push_back(v
);}else{q2
.push_back(val
);}}int popFront() {if(q1
.empty())return -1;if(sameLen()){ans
= q1
.front();q1
.pop_front();v
= q2
.front();q2
.pop_front();q1
.push_back(v
);}else{ans
= q1
.front();q1
.pop_front();}return ans
;}int popMiddle() {if(q1
.empty())return -1;ans
= q1
.back();q1
.pop_back();if(!sameLen() && !q2
.empty()){v
= q2
.front();q2
.pop_front();q1
.push_back(v
);}return ans
;}int popBack() {if(q1
.empty())return -1;if(sameLen()){ans
= q2
.back();q2
.pop_back();}else{if(!q2
.empty()){ans
= q2
.back();q2
.pop_back();v
= q1
.back();q1
.pop_back();q2
.push_front(v
);}else{ans
= q1
.back();q1
.pop_back();}}return ans
;}
};
60 ms 20.7 MB C++
我的CSDN博客地址 https://michael.blog.csdn.net/
長按或掃碼關注我的公眾號(Michael阿明),一起加油、一起學習進步!
總結
以上是生活随笔為你收集整理的LeetCode 1670. 设计前中后队列(deque)的全部內容,希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯,歡迎將生活随笔推薦給好友。