剑指offer | 9 用两个栈实现队列

1 题目

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

2 思路及解析

思路1：只把元素存放到一个栈中，另一个栈用来存放元素和周转元素！这样问题就简单了。

• 增加元素由于队列的后进后出，所以需要放到栈的最底端，故转移再增加再移回来！
• 删除元素直接最上面弹出即可！

代码：

# -*- coding:utf-8 -*-
class Solution:
    # 先初始化
    
    def __init__(self):
        self.s1 = [] # 元素都放在s1中！
        self.s2 = [] # s2起一个过渡的作用
        
    def push(self, node):
        # 逻辑：在s1中添加元素 队列需要后入后出 所以要将新插入的元素放到最底端！
        # 那怎么办呢？首先把s1中元素全部转移到s2中 然后插入到s1 再将s2中元素重新转移到s1
        while self.s1:
            # 即s1非空时 将s1的元素全部压到s2中
            self.s2.append(self.s1.pop())
        self.s1.append(node) # s1再添加元素！
        while self.s2:
            # 即s2非空时 将s2中元素全部压到s1中
            self.s1.append(self.s2.pop())
        
    def pop(self):
        # 元素都在s1中 所以删除直接从最上面弹出来即可！
        return self.s1.pop()

结果：

思路2：让 stack1 成为入队的！stack2 是出队的！ 不仅仅是思路1一样完全把另一个栈给空出来，而是充分利用两个栈的空间，而不是仅仅一个！

• 插入元素：直接在stack1append就可以了，因为stack1 是入队的！所以直接后面加就可以了
• 删除元素：最早进入的则最早被删除，如果stack2为空，需要把stack1元素压到stack2，然后从stack2弹出；如果stack2非空，直接将stack2的顶部元素弹出就好！
• 场景：一开始两个栈都是空的，然后全部往stack1里面加入了元素1 2 3 4 5，这时候我需要删除，按照队列的规则应该是先删除1，于是先把stack1全部压到stack2，然后从stack2弹出即可；ok，接下来，再继续加入元素，那么直接往stack1再append就好了，如果要删除直接就删除stack2，为空就先把stack1压到stack2再删，完美！

代码：

# -*- coding:utf-8 -*-
class Solution:
    # 先初始化
    
    def __init__(self):
        self.stack1 = [] # 入队
        self.stack2 = [] # 出队
    def push(self, node):
        # 直接往stack1中添加node
        self.stack1.append(node)
    def pop(self):
        if len(self.stack2) == 0:
            # 如果stack2为空 则 将stack1的所有元素压到stack2里面
            for _ in range(len(self.stack1)):
                self.stack2.append(self.stack1.pop())
            if len(self.stack2) == 0:
                # 如果此时stack2还为空 说明没有元素进来 即stack1为空 所以队列为空
                print("Queue is empty!")
                return # 返回空
        return self.stack2.pop() # 删除则删掉stack2最上面的！

结果：