设计前中后队列

105 阅读3分钟

要求

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

思路

可以将队列分成两个队列,left队列和right队列,这样就可以进行前中后的操作了

代码

var FrontMiddleBackQueue = function () {
    // 定义两个数组,方便进行前中后操作
    this.leftArray = [];
    this.rightArray = [];
};

/** 
 * @param {number} val
 * @return {void}
 */
FrontMiddleBackQueue.prototype.pushFront = function (val) {
    // 在左列表的最前面添加val
    this.leftArray.unshift(val);
    // 当左列表比右列表长,左列表最后一个数据给后列表
    if (this.leftArray.length > this.rightArray.length) {
        this.rightArray.unshift(this.leftArray.pop())
    }
};

/** 
 * @param {number} val
 * @return {void}
 */
FrontMiddleBackQueue.prototype.pushMiddle = function (val) {
    // 当左列表的长度跟右列表的长度一样,在右列表的表头添加数据(把右列表当成主列表)
    if (this.leftArray.length == this.rightArray.length) {
        this.rightArray.unshift(val);
    } else {
        // 左列表比右列表短时,添加到左列表的最后
        this.leftArray.push(val);
    }
};

/** 
 * @param {number} val
 * @return {void}
 */
FrontMiddleBackQueue.prototype.pushBack = function (val) {
    // 在右队列最后面添加元素
    this.rightArray.push(val);
    // 当右列表比左列表的长度长2以上,把右列表的第一个元素添加到左列表的最后
    if (this.rightArray.length > this.leftArray.length + 1) {
        this.leftArray.push(this.rightArray.shift())
    }
};

/**
 * @return {number}
 */
FrontMiddleBackQueue.prototype.popFront = function () {
    // 因为右列表是主列表,所以当右列表都没有长度时,表示全都没有值,返回-1
    if (!this.rightArray.length) return -1;
    // 当左列表没有长度时,返回右列表的第一个元素并剔除掉
    if (!this.leftArray.length) return this.rightArray.shift();
    // 把左列表的第一个元素保存起来并剔除掉
    let ret = this.leftArray.shift();
    // 然后判断左右列表的长度
    if (this.rightArray.length > this.leftArray.length + 1) {
        // 当右列表的长度比左列表的长度长2以上,把右列表的第一个元素添加到左列表的最后
        this.leftArray.push(this.rightArray.shift());
    }
    return ret;
};

/**
 * @return {number}
 */
FrontMiddleBackQueue.prototype.popMiddle = function () {
    // 因为右列表是主列表,所以当右列表都没有长度时,表示全都没有值,返回-1
    if (!this.rightArray.length) return -1;
    // 当左列表没有长度,表示只有右列表有一个数据,返回右列表的第一个数据
    if (!this.leftArray.length) return this.rightArray.shift();
    // 当左列表长度等于右列表长度,返回左列表的最后一个数据(右列表是主列表)
    if (this.leftArray.length == this.rightArray.length) {
        return this.leftArray.pop();
    }
    // 当右列表的长度大于左列表,返回右列表的第一个数据
    return this.rightArray.shift();
};

/**
 * @return {number}
 */
FrontMiddleBackQueue.prototype.popBack = function () {
    // 因为右列表是主列表,所以当右列表都没有长度时,表示全都没有值,返回-1
    if (!this.rightArray.length) return -1;
    // 把右列表的最后一个数据存起来并剔除掉
    let ret = this.rightArray.pop();
    // 判断长度,当左列表长度比右列表大时,把左列表最后一个元素剔除掉并且赋给右列表,当右列表的第一个元素
    if (this.leftArray.length > this.rightArray.length) {
        this.rightArray.unshift(this.leftArray.pop())
    }
    return ret;
};