美文网首页
剑指offer:05 用两个栈实现队列

剑指offer:05 用两个栈实现队列

作者: 毛毛毛毛毛豆 | 来源:发表于2019-08-07 16:23 被阅读0次

题目描述

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

基本思想

两个栈s1和s2,入队就将元素入s1栈,出队时,如果s2中为空,将s1里全部元素弹出压入s2,再弹出;如果s2中不为空,则弹出,直到s2为空

Python

class Solution:

    def __init__(self):

        self.s1 = []

        self.s2 = []

    def push(self, node):

        # write code here

        self.s1.append(node)

    def pop(self):

        # return

        if self.s2: #s2不为空

            return self.s2.pop()

        elif not self.s1: #s1为空

            return None

        else:

            while self.s1: # s2为空,s1不为空

                self.s2.append(self.s1.pop())

            return self.s2.pop()

相关文章

网友评论

      本文标题:剑指offer:05 用两个栈实现队列

      本文链接:https://www.haomeiwen.com/subject/lpfodctx.html