#yyds干货盘点# LeetCode程序员面试金典:三合一
精选
原创
©著作权归作者所有:来自51CTO博客作者灰太狼_cxh的原创作品,请联系作者获取转载授权,否则将追究法律责任
题目:
三合一。描述如何只用一个数组来实现三个栈。
你应该实现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; //判断当前栈是否已空
}
}