三合一 (如何只用一个数组来实现三个栈)

题目:

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

你应该实现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 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

 

你可能感兴趣的:(数据结构,c++,leetcode,职场和发展)