剑指offer 面试题9 （用两个栈实现队列） python

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

 

代码（）：

# -*- coding:utf-8 -*-

class Solution:

    def __init__(self):

        self.stack_A = []

        self.stack_B = []

    def push(self, node):

        # write code here

        self.stack_A.append(node)

    def pop(self):

        if (not self.stack_B) and (not self.stack_A):

            return None

        elif self.stack_B:

            return self.stack_B.pop()

        else:

            while self.stack_A:

                self.stack_B.append(self.stack_A.pop())

            return self.stack_B.pop()

解题思路：

栈是先进后厨FILO

队列是先进先出FIFO

这里有两个栈 A  B  

那么想法就是 先把东西放进A 然后如果B为空就把A里面的放进B。

例如：原数列为 1 2 3 4 5  进入A后变为 5 4 3 2 1 再放进B就变成 1 2 3 4 5

如果B不为空，则先把B里面的先输出，输出完毕后再填补A的数据进去