剑指offer-用两个栈实现队列

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

思路：栈是先进后出，队列是先进先出。将栈A作为入队列，栈B作为出队列。

栈A用来作入队列

栈B用来出队列，当栈B为空时，栈A全部出栈（将元素入栈的顺序翻转了）到栈B,栈B再出栈（即出队列，顺序就变成了先进先出了）

class Solution:
    def __init__(self):
        self.stack1 = []
        self.stack2 = []

    def push(self, node):
        # write code here
        self.stack1.append(node)

    def pop(self):
        # 栈1负责存储数据，栈2负责弹出数据，每次弹出后，栈2要将数据返还给栈1，所以栈2肯定为空，栈1不为空
        if len(self.stack2) != 0 or len(self.stack1) == 0:
            return -1
        else:
            # 将栈1数据压到栈2
            while len(self.stack1) >= 1:
                self.stack2.append(self.stack1.pop())

            # 待返回数据
            res = self.stack2.pop()

            # 将栈2剩余数据返还给栈1
            while len(self.stack2):
                self.stack1.append(self.stack2.pop())
            return res

# 测试代码
if __name__ == '__main__':
    s = Solution()
    s.push(1)
    s.push(2)
    s.push(3)
    print(s.pop())
    s.push(4)
    print(s.pop())