面试题9：用两个栈实现队列

这一题还是挺难的，有两个栈stackA、stackB，A是入栈的，B是出栈的，入栈时，直接进入A即可，出栈时，先判断是否有元素，

如果B没有元素，pop肯定报错，应该先将A中所有的元素压倒B里面，再pop最上面一个元素，如果B中有就直接pop出，就可以，

这是最优的思路，两个栈实现了先进后出，在一起又实现了队列的先进先出。

再次整理：

　　　　1.首先init2个栈，一个做数据的输入栈A，一个做数据的输出栈B

　　　　2.数据压入的时候，直接栈A.append节点

　　　　3.数据输出的时候，需要考虑，这个时候的栈B是否为空，分两种情况，a.不为空，直接pop()； b.为空，那就在栈A不为空的情况下将栈A数据压入栈B,压完所有数据，就返回.pop()

 

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stackA = []
        self.stackB = []
    def push(self, node):
        # write code here
        self.stackA.append(node)
    def pop(self):
        if self.stackB != []:
            return self.stackB.pop()
        if self.stackB == []:
            while self.stackA:
                self.stackB.append(self.stackA.pop())
            return self.stackB.pop()

　　

转载于:https://www.cnblogs.com/ivyharding/p/11216045.html