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

题目:

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

你应该实现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();
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值