题解 | #用两个栈实现队列#

用两个栈实现队列

http://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6

维持两个栈:stack1负责插入,stack2负责删除

  • 当进队列时,直接插入stack1;

  • 当出队列时,

    若stack2为空,则将stack1的元素依次弹出到stack2

    若stack2不为空,则弹出stack2栈顶元素

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack1 = []  # 负责插入
        self.stack2 = []  # 负责删除
    def push(self, node):
        self.stack1.append(node)
        
    def pop(self):
        # stack2为空时,将stack1元素弹到stack2
        if not self.stack2:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
        # 弹出stack2栈顶元素
        return self.stack2.pop()
                
全部评论

相关推荐

07-01 17:14
中北大学 Java
兄弟们是真是假
牛客46374834...:我在boss上投java岗从来没成功过
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务