面试题 03.01. 三合一

面试题 03.01. 三合一

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

你应该实现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`不压入元素。

题解

class TripleInOne {

    int[] stack;
    int stackSize;
    int[] nums = new int[]{-1, -1, -1};

    public TripleInOne(int stackSize) {
        stack = new int[stackSize * 3];
        this.stackSize = stackSize;
    }
    
    public void push(int stackNum, int value) {
        if (nums[stackNum] + 1 == stackSize) return;
        nums[stackNum] += 1;
        stack[nums[stackNum] + stackNum * stackSize] = value;
    }
    
    public int pop(int stackNum) {
        int res = peek(stackNum);
        if (res != -1) nums[stackNum] -= 1;
        return res;
    }
    
    public int peek(int stackNum) {
        if (nums[stackNum] == -1) {
            return -1;
        } else {
            int res = stack[nums[stackNum] + stackNum * stackSize];
            return res;
        }
    }
    
    public boolean isEmpty(int stackNum) {
        return nums[stackNum] == -1;
    }
}