题目:
三合一。描述如何只用一个数组来实现三个栈。
你应该实现push(stackNum, value)
、pop(stackNum)
、isEmpty(stackNum)
、peek(stackNum)
方法。stackNum
表示栈下标,value
表示压入的值。
构造函数会传入一个stackSize
参数,代表每个栈的大小。
输入:
["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`不压入元素。
示例解释:
1.执行TripleInOne(1)函数,意思是每个栈的容量为1,无返回值,所以输出为null
2.执行push(0,1)函数,意思是向下标为0的栈中将1入栈,无返回值,所以输出为null
3.执行push(0,2)函数,意思是向下标为0的栈中将2入栈,无返回值,所以输出为null
4.执行pop(0)函数,意思是将下标为0的栈顶元素出栈,由于栈的容量为1,此时栈的栈顶元素为1,所以返回1
5.执行pop(0)函数,意思是将下标为0的栈顶元素出栈,前面已经将栈顶元素出栈了,此时栈为空,所以返回-1
6.与5同理
7.执行isEmpty(0)函数,意思是判断下标为0的栈是否为空,此时为空,返回true
源代码如下:
class TripleInOne {
public:
//用数组模拟栈
vector<int> res[3];//定义一个三行的二维数组,一行表示一个栈
int stackSize;//栈的大小
TripleInOne(int stackSize) {
this->stackSize=stackSize;
}
void push(int stackNum, int value) {
//res[stackNum]表示第stackNum+1个栈,因为下标从0开始
int size=res[stackNum].size();//计算当前栈的元素个数
//若size小于栈的大小,才能将元素入栈(也就是放到数组中)
if(size<stackSize)
{
res[stackNum].push_back(value);
}
}
int pop(int stackNum) {
//先判断栈是否为空,也就是数组当前行是否为空
if(res[stackNum].empty()) return -1;
//保存栈顶元素
int value=res[stackNum].back();
//当前行的末尾元素删除
res[stackNum].pop_back();
//返回刚刚保存的栈顶元素的值
return value;
}
//peek()用于看接下来读入的那个字符是什么,如果是栈的话,那就只能是栈顶元素了
int peek(int stackNum) {
if(res[stackNum].empty()) return -1;
int value=res[stackNum].back();
return value;
}
bool isEmpty(int stackNum) {
//判断数组中当前这一行是否为空,即栈是否为空
return res[stackNum].empty();
}
};