09用两个栈实现一个队列

用两个栈实现一个队列。队列的声明如下,请实现它的两个方法push和pop,分别完成在队列尾部插入结点和在队列头部删除结点的功能。

# -*- coding:utf-8 -*-
class QueueWithTwoStacks:
    def __init__(self):
        self.stack1 = []
        self.stack2 = []
        
    def push(self, node):
        self.stack1.append(node)

    def pop(self):
        if self.stack2:
            return self.stack2.pop()
        else:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
            if self.stack2:
                return self.stack2.pop()
    原文作者:黄健楸
    原文地址: https://www.jianshu.com/p/b7d17ea98211
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞