JAVA练习135-三合一

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

你应该实现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

分析：

方法：数组分段

用一个数组实现三个栈，可以将数组分成三部分，分别代表不同的栈，但是要进行栈操作，就需要直到每个栈的当前索引，因此我们可以将数组再扩大 3 的长度分别代表三个栈的当前索引，还可以再扩大 1 的长度来存储栈的大小，这样删除时，对应索引值赋 0，索引减一，添加时对应索引值赋值，索引加一，注意判断大小就是了。

时间复杂度：O(n)
空间复杂度：O(n)

class TripleInOne {

    //创建数组
    int[] nums;

    public TripleInOne(int stackSize) {
        nums = new int[stackSize*3+4];
        //存储栈大小
        nums[3] = stackSize;
    }
    
    public void push(int stackNum, int value) {
        //栈不满时压入
        if(nums[stackNum] < nums[3]){
            nums[nums[3] * stackNum + 4 + nums[stackNum]++] = value;
        }
    }
    
    public int pop(int stackNum) {
        //栈为空
        if(nums[stackNum] == 0){
            return -1;
        }
        //记录元素
        int pop = nums[nums[3] * stackNum + 4 + nums[stackNum]-1];
        //删除元素
        nums[nums[3] * stackNum + 4 + --nums[stackNum]] = 0;
        return pop;
    }
    
    public int peek(int stackNum) {
        //栈为空
        if(nums[stackNum] == 0){
            return -1;
        }
        return nums[nums[3] * stackNum + 4 + nums[stackNum]-1];
    }
    
    public boolean isEmpty(int stackNum) {
        return nums[stackNum] == 0;
    }
}

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

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/three-in-one-lcci