用两个栈实现队列——python

题目描述：
用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

解题思路：
1.栈是先进后出的，队列是先进先出的，如下图。

2.把新添加进来的节点都添加到栈1上，然后栈1上的节点添加到栈2上，再将栈2上的节点输出，就是实现了队列的输出方式。输出pop（）函数具体可以分为以下三种情况。
（1）当栈2非空时，直接将栈2的最后一个节点输出（由于栈的特点时先进后出，后进先出的，故先输出的是最后一个节点）
（2）当栈1为空，栈2非空，将栈1内的节点从后往前逐个加入到栈2中，最后再输出删除掉栈2中最后一个节点
（3）当栈1、栈2都为空时，说明队列已经完成输出了

python代码如下：

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack1 = []
        self.stack2 = []
    
    def push(self, node):
        # write code here
        self.stack1.append(node)
        
        
    def pop(self):
        if self.stack2:##栈2非空,直接删掉栈2的最后一个节点
            return self.stack2.pop()
        elif self.stack1:##如果栈1，栈2都为空，算法结束
            """
            栈1为空，栈2非空，将栈1内的节点从后往前逐个加入到栈2中，
            最后再删掉栈2中最后一个节点
            """
            while self.stack1:
                self.stack2.append(self.stack1.pop())
            return self.stack2.pop()
        else:
            return 
           

ps：
列表 List pop(obj)函数用于移除列表中的一个元素（obj为可选参数，可以省略，默认最后一个元素），并且返回该元素的值。