40. 用栈实现队列

40. 用栈实现队列

中文English

正如标题所述，你需要使用两个栈来实现队列的一些操作。

队列应支持push(element)，pop() 和 top()，其中pop是弹出队列中的第一个(最前面的)元素。

pop和top方法都应该返回第一个元素的值。

样例

例1:

输入:
    push(1)
    pop()    
    push(2)
    push(3)
    top()    
    pop()     
输出:
    1
    2
    2

例2:

输入:
    push(1)
    push(2)
    push(2)
    push(3)
    push(4)
    push(5)
    push(6)
    push(7)
    push(1)
输出:
[]

挑战

仅使用两个栈来实现它，不使用任何其他数据结构，push，pop 和 top的复杂度都应该是均摊O(1)的

注意事项

假设调用pop()函数的时候，队列非空 

输入测试数据 (每行一个参数)如何理解测试数据？

class MyQueue:
    
    def __init__(self):
        # do intialization if necessary
        self.stack1 = []
        self.stack2 = []

    """
    @param: element: An integer
    @return: nothing
    """
    def push(self, element):
        # write your code here
        self.stack1.append(element)

    """
    @return: An integer
    """
    def pop(self):
        # write your code here
        if (not self.stack2):
            self.move()
        
        return self.stack2.pop()

    """
    @return: An integer
    """
    def top(self):
        # write your code here
        if (not self.stack2):
            self.move()
        
        return self.stack2[-1]
        
    
    def move(self):
        #如果栈2空，把栈1的全部移过去
        while (self.stack1):
            self.stack2.append(self.stack1.pop())