2022-5-16 设计前中后队列

设计前中后队列

题目

请你设计一个队列，支持在前，中，后三个位置的 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] 。

示例

输入：
["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 -> [] （队列为空）

代码

struct MyListNode {
    int val;
    MyListNode *prev;
    MyListNode *next;

    MyListNode(int val, MyListNode *prev = nullptr, MyListNode *next = nullptr) : val(val), prev(prev), next(next) {}
};

class FrontMiddleBackQueue {  // 两个链表实现版本
private:
    MyListNode *dummyHead1, *dummyTail1;
    MyListNode *dummyHead2, *dummyTail2;
    int size1,size2;

    void insertNode(MyListNode* nodeBefore,int val){
        auto node=new MyListNode(val,nodeBefore,nodeBefore->next);
        node->prev->next=node;
        node->next->prev=node;
    }

    void insertNode(MyListNode* nodeBefore,MyListNode* node){
        node->prev=nodeBefore;
        node->next=nodeBefore->next;
        node->prev->next=node;
        node->next->prev=node;
    }

    MyListNode* popNode(MyListNode *node){
        node->prev->next=node->next;
        node->next->prev=node->prev;
        return node;
    }

    void moveBack(){
        auto node=popNode(dummyTail1->prev);
        insertNode(dummyHead2,node);
        size1--;
        size2++;
    }
    
    void moveFront(){
        auto node=popNode(dummyHead2->next);
        insertNode(dummyTail1->prev,node);
        size2--;
        size1++;
    }

    bool check(){
        return (size1==size2) || (size1==size2+1);
    }

public:
    FrontMiddleBackQueue():size1(0), size2(0) { // size1==size2 or size1==size2+1
        dummyHead1=new MyListNode(0);
        dummyTail1=new MyListNode(0);
        dummyHead2=new MyListNode(0);
        dummyTail2=new MyListNode(0);
        dummyHead1->next=dummyTail1;
        dummyTail1->prev=dummyHead1;
        dummyHead2->next=dummyTail2;
        dummyTail2->prev=dummyHead2;
    }
    
    void pushFront(int val) {
        insertNode(dummyHead1,val);
        size1++;
        if(!check()) moveBack();
    }
    
    void pushMiddle(int val) {
        if(size1!=size2) moveBack();
        insertNode(dummyTail1->prev,val);
        size1++;
    }
    
    void pushBack(int val) {
        insertNode(dummyTail2->prev,val);
        size2++;
        if(!check()) moveFront();
    }
    
    int popFront() {
        if(size1==0) return -1;
        auto node=popNode(dummyHead1->next);
        size1--;
        if(!check()) moveFront();     
        return node->val;
    }
    
    int popMiddle() {
        if(size1==0) return -1;
        auto node=popNode(dummyTail1->prev);
        size1--;
        if(!check()) moveFront();
        return node->val;
    }
    
    int popBack() {
        if(size2==0 && size1==0) return -1;
        if(size2==0 && size1!=0) return popMiddle();
        auto node=popNode(dummyTail2->prev);
        size2--;
        if(!check()) moveBack();
        return node->val;
    }
};