LeetCode 算法：三合一

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第 4 天，点击查看活动详情

三合一

原题地址

三合一。描述如何只用一个数组来实现三个栈。

你应该实现 push(stackNum, value)、pop(stackNum)、isEmpty(stackNum)、peek(stackNum)方法。stackNum 表示栈下标，value 表示压入的值。

构造函数会传入一个 stackSize 参数，代表每个栈的大小。

示例1:

 输入：
["TripleInOne", "push", "push", "pop", "pop", "pop", "isEmpty"]
[[1], [0, 1], [0, 2], [0], [0], [0], [0]]
 输出：
[null, null, null, 1, -1, -1, true]
说明：当栈为空时`pop, peek`返回-1，当栈满时`push`不压入元素。

示例2:

输入：
["TripleInOne", "push", "push", "push", "pop", "pop", "pop", "peek"]
[[2], [0, 1], [0, 2], [0, 3], [0], [0], [0], [0]]
 输出：
[null, null, null, null, 2, 1, -1, -1]

提示：

• 0 <= stackNum <= 2

思路分析

1. 读了好几遍题目总算是把题目看懂了，意思是有三个栈，然后入参中的 stackNum 表示要操作的是第几个栈；
2. 理解题目后，使用数组来表示栈，就会显得比较简单了；
3. 首先实现初始化 TripleInOne，可以定义一个二维数组中包含三个一维数组来表示这三个栈，给到的 stackSize 即为栈的长度；
4. push 操作时，需要先判断对应栈的长度有没有超过初始化中的总长度，若没有超过再将元素 push 到对应的栈中；
5. pop 操作，若对应栈的长度为 0 时，则返回 -1；否则，返回对应数组 pop 的值；
6. peek 操作，数组中没有 peek 方法，而我们了解的 peek 操作跟 pop 操作的区别就是 peek 操作不会改变原数组，它们都会返回数组的最后一个元素，因此需要返回数组的最后一个元素；
7. 判断对应数组的长度是否为 0 即可知道对应的栈是否为空。

AC 代码

/**
 * @param {number} stackSize
 */
var TripleInOne = function(stackSize) {
    this.stack = [[], [], []]
    this.size = stackSize
};

/** 
 * @param {number} stackNum 
 * @param {number} value
 * @return {void}
 */
TripleInOne.prototype.push = function(stackNum, value) {
    if(this.stack[stackNum].length >= this.size) return
    this.stack[stackNum].push(value)
};

/** 
 * @param {number} stackNum
 * @return {number}
 */
TripleInOne.prototype.pop = function(stackNum) {
    const len = this.stack[stackNum].length
    if(!len) return -1
    else return this.stack[stackNum].pop()
};

/** 
 * @param {number} stackNum
 * @return {number}
 */
TripleInOne.prototype.peek = function(stackNum) {
    const len = this.stack[stackNum].length
    if(!len) return -1
    else return this.stack[stackNum][len - 1]
};

/** 
 * @param {number} stackNum
 * @return {boolean}
 */
TripleInOne.prototype.isEmpty = function(stackNum) {
    return this.stack[stackNum].length === 0
};

/**
 * Your TripleInOne object will be instantiated and called as such:
 * var obj = new TripleInOne(stackSize)
 * obj.push(stackNum,value)
 * var param_2 = obj.pop(stackNum)
 * var param_3 = obj.peek(stackNum)
 * var param_4 = obj.isEmpty(stackNum)
 */

结果：

• 执行结果： 通过
• 执行用时：168 ms, 在所有 JavaScript 提交中击败了54.84%的用户
• 内存消耗：55.1 MB, 在所有 JavaScript 提交中击败了91.40%的用户
• 通过测试用例：30 / 30

END