(剑指offer)面试题9:用两个栈实现队列

栈后进先出,队列先进先出。
1.当stack2不为空时,stack2中的栈顶元素是最先进入队列的元素,可以弹出。
2.当stack2为空时,把stack1中的元素逐个弹入并压入stack2中,先入stack1的元素这时在stack2顶部,这时弹出stack2就完成了队列
(书上的图更便于理解)

class Solution:
    def __init__(self):
        self.stack1=[]
        self.stack2=[]
    def push(self,node):
        self.stack1.append(node) # 加入,一直从stack1添加
    def pop(self):
        if len(self.stack2)==0 and len(self.stack1)==0:
            return
        elif len(self.stack2)==0:
            while len(self.stack1)>0:  # 把stack1的元素全部弹出,并压入stack2
                self.stack2.append(self.stack1.pop())
        return self.stack2.pop()  # 把stack2中的栈顶元素弹出,即完成了队列


# s = Solution()
# s.push(1)
# s.push(2)
# s.push(3)
# s.push(4)
# print(s.pop())
# s.push(5)
# print(s.pop())
# print(s.pop())
# print(s.pop())
# s.push(6)
# print(s.pop())
# print(s.pop())
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值