【leetcode】面试题 03.01. 三合一

一、题目

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

你应该实现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);
 */

四、运行结果