#yyds干货盘点# LeetCode程序员面试金典：三合一

题目：

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

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

代码实现：

class TripleInOne {
    private final int STACK_SIZE;   //栈大小
    private int[] index;    //当前栈的栈顶指针
    private int[][] stack;  //栈

    public TripleInOne(int stackSize) {     //初始化操作
        this.STACK_SIZE=stackSize;
        this.index=new int[]{-1,-1,-1};
        this.stack=new int[3][STACK_SIZE];
    }

    public void push(int stackNum, int value) {
        if(index[stackNum]+1<STACK_SIZE){     //判断当前栈是否已满
            stack[stackNum][++index[stackNum]]=value;     //向栈内添加元素
            //index[stackNum]++;      //栈顶指针上移一位
        }
    }

    public int pop(int stackNum) {
        if(isEmpty(stackNum)){    //判断当前栈是否已空
            return -1;
        }
        //index[stackNum]--;      //将栈顶指针向下移一位
        return stack[stackNum][index[stackNum]--];    //返回抛出的元素
    }

    public int peek(int stackNum) {
        if(isEmpty(stackNum)){    //判断当前栈是否已空
            return -1;
        }
        return stack[stackNum][index[stackNum]];      //返回栈顶元素
    }

    public boolean isEmpty(int stackNum) {
        return index[stackNum]==-1;     //判断当前栈是否已空
    }
}