python--使用两个栈实现队列

问题:用两个栈来实现一个队列,完成队列的PushPop操作。 

 分析:栈的特性是“先进后出”,队列为“先进先出”。

python--使用两个栈实现队列_第1张图片python--使用两个栈实现队列_第2张图片

思路:入队:直接把新元素压入stack1即可。 

           出队:根据队列先进先出的性质,由于先进入队列的元素被压倒stack1的栈底,要想实现先入队列的先出队,需要将stack1中的元素逐个弹出并压入stack2,经过弹出和压入之后最先进入的元素就处于stack2的栈顶,有可以直接弹出。

python实现代码:

class Solution:
    def __init__(self):
        # 初始化两个空栈
        self.stack1 = []
        self.stack2 = []
    # 实现入队操作
    def push(self, node):
        # 直接压入 栈1
        self.stack1.append(node)
    
    # 实现出队操作
    def pop(self):
        if len(self.stack2) == 0:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
        return self.stack2.pop()

 

你可能感兴趣的:(栈,队列,python,数据结构,python)