一、题目
三合一。描述如何只用一个数组来实现三个栈。
你应该实现push(stackNum, value)、pop(stackNum)、isEmpty(stackNum)、peek(stackNum)方法。stackNum表示栈下标,value表示压入的值。
构造函数会传入一个stackSize参数,代表每个栈的大小。
二、解题思路
刚开始看题目有点晦涩难懂,研究了半天,原来so eash,就是将三个栈合并到一个数组里了,也就是下标(x,x+1,x+2)分别表示栈1、栈2、栈3.
栈1下标:0,3,6,9,12,.....
栈2下标:1,4,7,10,13,.....
栈1下标:2,5,8,11,14,.....
题目中的stackNum表示对第几个栈进行操作。与普通栈相比,只需要top指针每次操作时+或-3就行了。
刚开始,我们分别初始化栈顶指针为-3,-2,-1。因此想要判断栈是否为空,只要判断栈顶指针是否<0便可。
三、代码
class TripleInOne {
int nums[];//记录栈
int top[];//记录三个栈顶指针位置
int stackSize;//记录栈的大小
public TripleInOne(int stackSize) {
this.nums=new int[stackSize*3];
this.top= new int[]{-3,-2,-1};//初始化
this.stackSize=stackSize*3;
}
public void push(int stackNum, int value) {
if(this.top[stackNum]+3 < this.stackSize) {
this.top[stackNum]+=3;//每次添加一个元素,栈顶指针下标需要+3
nums[this.top[stackNum]]=value;
}
}
public int pop(int stackNum) {
//需要判断该栈stackNum是否为空
if(this.top[stackNum] >=0) {
int ans=nums[this.top[stackNum]];
this.top[stackNum]-=3;
return ans;
}
return -1;
}
public int peek(int stackNum) {
if(this.top[stackNum] >=0) {
return nums[this.top[stackNum]];
}
return -1;
}
public boolean isEmpty(int stackNum) {
//栈顶指针<0即为空栈
return this.top[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);
*/
四、运行结果