快乐的LeetCode --- 面试题09. 用两个栈实现队列

题目描述:

用两个栈来实现一个队列,完成队列的Push和Pop操作。 若队列中没有元素,返回-1


解题思路:

   栈的性质是先进后出,队列的性质是先进先出,队列的push操作,可以使用一个栈来完成,队列的pop操作,需要把一个栈的数据转到另外一个栈上,然后只需要对另外一个栈,执行pop()操作即可,因为此时的栈顶即是第一个栈的栈低,看成队列的话,即是队列的出口处。


代码1:

class Solution
{
public:
    void push(int node) 
    {
        stack1.push(node);
    }

    int pop() 
    {
        while(!stack1.empty())
        {
            stack2.push(stack1.top());
            stack1.pop();
        }
        int num=stack2.top();
        stack2.pop();
        while(!stack2.empty())
        {
            stack1.push(stack2.top());
            stack2.pop();
        }
        return num;
    }
private:
    stack<int> stack1;
    stack<int> stack2;
};

代码2:

class Solution:
    def __init__(self):
        self.stack1 = []
        self.stack2 = []

    def push(self, node):
        self.stack1.append(node)

    def pop(self):
        # 检查stack2是否为空,不为空弹出,
        # 为空继续判断stack1是否为空,不为空放入stack2中,为空表示栈内无元素
        try:
            _pop = self.stack2.pop()
        except IndexError as e:
            if len(self.stack1) != 0:
                for i in range(len(self.stack1)):
                    self.stack2.append(self.stack1.pop())
                _pop = self.stack2.pop()
            else:
                _pop = -1
        return _pop

if __name__ == '__main__':
    s = Solution()
    s.push(1)
    s.push(2)
    s.push(3)
    s.push(4)
    print(s.pop())
    print(s.pop())
    print(s.pop())
    print(s.pop())
    s.push(5)
    print(s.pop())
    print(s.pop())

代码3:

class CQueue:
    def __init__(self):
        self.A, self.B = [], []

    def appendTail(self, value):
        self.A.append(value)

    def deleteHead(self):
        if self.B: return self.B.pop()
        if not self.A: return -1
        while self.A:
            self.B.append(self.A.pop())
        return self.B.pop()

题目来源:

面试题09. 用两个栈实现队列


参考链接:

https://blog.csdn.net/weixin_42788078/article/details/88565550

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值