栈和队列---由两个栈组成队列

【题目】

  编写一个类,用两个栈实现队列,支持队列的基本操作(add, poll, peek)。

【基本思路】

  使用两个栈stackpush、stackpop,stackpush栈负责压入数据、stackpop栈负责将stackpush中的元素逆序,用于获取或者弹出栈顶元素。但是有一个规则:stackpop只有为空的时候才再次向stackpush栈索要元素,而且,必须一次拿走stackpush中当前的所有元素。

【代码实现】

#python3.5
class TwoStackQueue:
    stackPush = []
    stackPop = []

    def add(self, newNum):
        self.stackPush.append(newNum)

    def poll(self):
        if not self.stackPush and not self.stackPop:
            raise Exception("Queue is empty!")
        elif not self.stackPop:
            while self.stackPush:
                self.stackPop.append(self.stackPush.pop())
        return self.stackPop.pop()

    def peek(self):
        if not self.stackPush and not self.stackPop:
            raise Exception("Queue is empty!")
        elif not self.stackPop:
            while self.stackPush:
                self.stackPop.append(self.stackPush.pop())
        return self.stackPop[-1]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值