[路飞]设计前中后队列

81 阅读2分钟

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

本题解题思路

可以将该队列查分成两个数组来处理,使其左边数组一直保持加一等于右边数组,或者等于右边数组,始终保持左边数组小于等于右边数组代码如下:

var FrontMiddleBackQueue = function() {
    //左边数组
    this.arrL = [];
    //右边数组
    this.arrR = [];
};

/** 
 * @param {number} val
 * @return {void}
 */
FrontMiddleBackQueue.prototype.pushFront = function(val) {
    //在左边数组头部添加节点
    this.arrL.unshift(val);
    //判断左边数组是否大于右边数组,如果是就将左边末尾的节点加到右边数组开头
    if(this.arrL.length > this.arrR.length){
        this.arrR.unshift(this.arrL.pop())
    }
};

/** 
 * @param {number} val
 * @return {void}
 */
FrontMiddleBackQueue.prototype.pushMiddle = function(val) {
   //先将节点接到左边数组末尾
   this.arrL.push(val);
   //保持左边数组小于等于右边数组
   if(this.arrL.length > this.arrR.length){
       this.arrR.unshift(this.arrL.pop())
   }
};

/** 
 * @param {number} val
 * @return {void}
 */
FrontMiddleBackQueue.prototype.pushBack = function(val) {
   //右边节点末尾现在节点
   this.arrR.push(val);
   //需要始终保持右边节点大于左边节点一个或者等于左边节点,所以如果右边节点大于左边一个以上需要右边节点将头节点给到左边节点末尾
    if(this.arrL.length + 1 < this.arrR.length){
        this.arrL.push(this.arrR.shift());
    }
};

/**
 * @return {number}
 */
FrontMiddleBackQueue.prototype.popFront = function() {
    //因为右边数组始终大于左边节点,所以如果右边数组不存在那么改队列就是空队列
    if(!this.arrR.length) return -1;
    //如果左边数组不存在那么右边数组就只有一个节点,所以只能删除右数组
    if(!this.arrL.length) return this.arrR.pop();
    //左数组存在那么删除左数组头结点
    let frontValue =  this.arrL.shift();
    //平衡两边数组,保持左边数组小于或者等于右边数组长度
    if(this.arrL.length + 1 < this.arrR.length){
        this.arrL.push(this.arrR.shift());
    }
    return frontValue;
};

/**
 * @return {number}
 */
FrontMiddleBackQueue.prototype.popMiddle = function() {
   //因为右边数组始终大于左边节点,所以如果右边数组不存在那么改队列就是空队列
    if(!this.arrR.length) return -1;
    //当左右两边数组长度相等时,删左边数组尾结点,否则删右边数组头结点
    if(this.arrL.length == this.arrR.length){
        return this.arrL.pop();
    }else {
        return this.arrR.shift()
    }
};

/**
 * @return {number}
 */
FrontMiddleBackQueue.prototype.popBack = function() {
 //因为右边数组始终大于左边节点,所以如果右边数组不存在那么改队列就是空队列
    if(!this.arrR.length) return -1;
    //删除右节点尾结点
      let backValue = this.arrR.pop();
        //平衡两边数组,保持左边数组小于或者等于右边数组长度
      if(this.arrL.length > this.arrR.length){
         this.arrR.unshift(this.arrL.pop());
    }
    return backValue;
};