面试题 03.01. 三合一
三合一。描述如何只用一个数组来实现三个栈。
你应该实现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`不压入元素。
题解
class TripleInOne {
int[] stack;
int stackSize;
int[] nums = new int[]{-1, -1, -1};
public TripleInOne(int stackSize) {
stack = new int[stackSize * 3];
this.stackSize = stackSize;
}
public void push(int stackNum, int value) {
if (nums[stackNum] + 1 == stackSize) return;
nums[stackNum] += 1;
stack[nums[stackNum] + stackNum * stackSize] = value;
}
public int pop(int stackNum) {
int res = peek(stackNum);
if (res != -1) nums[stackNum] -= 1;
return res;
}
public int peek(int stackNum) {
if (nums[stackNum] == -1) {
return -1;
} else {
int res = stack[nums[stackNum] + stackNum * stackSize];
return res;
}
}
public boolean isEmpty(int stackNum) {
return nums[stackNum] == -1;
}
}