天天看點

#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`不壓入元素。      
輸入:
["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;     //判斷目前棧是否已空
    }
}