題目:
三合一。描述如何隻用一個數組來實作三個棧。
你應該實作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; //判斷目前棧是否已空
}
}