【剑指offer】5.用两个栈实现队列（Python）

知识点：栈、队列

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

• 栈无法实现队列功能： 栈底元素（对应队首元素）无法直接删除，需要将上方所有元素出栈。
• 双栈可实现列表倒序： 设有含三个元素的栈 stack1 = [1,2,3]和空栈 stack2 = []。若循环执行 stack1元素出栈并添加入栈 stack2，直到栈 stack1为空，则 stack1= [], stack2 = [3,2,1]，即 栈 stack2 元素实现栈 stack1 元素倒序 。
• 利用栈 stack2 删除队首元素： 倒序后，stack2 执行出栈则相当于删除了 stack1 的栈底元素，即对应队首元素。

函数设计：

• 加入队尾pop()函数： 将数字 val 加入栈 stack1 即可
• 删除队首push()函数： 有以下三种情况：

（1）当栈 stack2 不为空： stack2中仍有已完成倒序的元素，因此直接返回 stack2 的栈顶元素。
（2）否则：当 stack1 为空： 即两个栈都为空，无元素，因此返回 -1 。
（3）否则： 将栈 stack1 元素全部转移至栈stack2 中，实现元素倒序，并返回栈 stack2 的栈顶元素。

# -*- 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):
        # return xx
        if self.stack2:
            return self.stack2.pop()
        else:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
            return self.stack2.pop()

'''
用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。
'''
# -*- coding:utf-8 -*-
class Solution:
    def __init__(self,lst1,lst2):
        self.stack1=lst1
        self.stack2=lst2
    #使用栈，队列尾部添加元素,直接使用append()函数添加元素

    def push(self, node):
        # write code here
        print("初始队列为："+str(self.stack1))
        self.stack1.append(node)
        print("添加元素"+str(node)+"后为："+str(self.stack1))

    #使用栈，删除队列元素，即删除队列头部元素，如果栈2不为空，则栈2删除元素直到栈2为空；
    #此时将栈1的元素一个一个添加到栈2中，最后将栈2中的元素一个一个抛出。
    def pop(self):
        print("辅助栈2的初始元素有："+str(self.stack2))
        # return xx
        if self.stack2:
            st2=self.stack2.pop()
            print("此次栈2的pop操作删除的元素为："+str(st2))
        else:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
            print("将栈1的所有元素添加到栈2中，此时栈2的元素有："+str(self.stack2))
            st1=self.stack2.pop()
            print("此次栈2的pop操作删除的元素为："+str(st1))

s=Solution([1,3,5],[2,4])
s.push(7)
s.push(9)
s.pop()
s.pop()
s.pop()