JZ5 用两个栈实现队列

描述

用两个栈来实现一个队列，分别完成在队列尾部插入整数(push)和在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法，即保证pop操作时队列内已有元素。

示例:

输入: ["PSH1","PSH2","POP","POP"]

返回: 1,2

解析

"PSH1":代表将1插入队列尾部

"PSH2":代表将2插入队列尾部

"POP“:代表删除一个元素，先进先出=>返回1

"POP“:代表删除一个元素，先进先出=>返回2

栈的特性是先进后出，队列的特性是先进先出。

没有想法来着，看了讨论组的题解，大概意思就是一个栈充当进队入口，一个栈充当出队入口，两个栈之间再进行数据交换。

Stack1先将元素进栈，如1，2，3，也就是push的操作。

pop操作主要由Stack2实现，首先如果Stack2为空(也就是还没有元素输入或者所有输入元素均输出)的情况，这时候按Stack1出栈的元素顺序存入Stack2，也就是Stack1从3开始pop，Stack2从3开始push。

当元素存完后，最先输入到Stack1的1变成了Stack2的最顶部，这时候执行Stack2的pop操作，也就是出栈，就是从1开始输出了，相当于实现了队列的先进先出特性。

class Solution:
    def __init__(self):
        self.Stack1 = []
        self.Stack2 = []
    def push(self, node):
        self.Stack1.append(node)
    def pop(self):
        if self.Stack2 == []:
            while self.Stack1:
                self.Stack2.append(self.Stack1.pop())
        if self.Stack2 != []:
            return self.Stack2.pop()
        else:
            return None