python基本语法3.11-用栈实现队列

问题描述:用栈(使用list)实现队列:支持push(),pop(),top()方法。pop和top方法都应该返回第一个元素的值.

 

class MyQueue:
    def __init__(self):
        self.stack_1 = []
        self.stack_2 = []

    def push(self,element):
        while self.stack_1:
            self.stack_2.append(self.stack_1.pop())
        self.stack_1.append(element)
        while self.stack_2:
            self.stack_1.append(self.stack_2.pop())
    def top(self):
        if self.stack_1:
            return self.stack_1[-1]
    def pop(self):
        return self.stack_1.pop()

queue = MyQueue()
queue.push(1)
print queue.pop()
queue.push(2)
queue.push(3)
print queue.top()
print queue.pop()

 

posted on 2017-10-26 16:01  小嘤嘤  阅读(230)  评论(0编辑  收藏  举报

导航