(python version) 剑指offer 9.用两个栈实现队列

题目描述

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

解题笔记

• 栈（Stack）：后进先出
• 队列（Queue）：先进先出
• 过程
  • 判断 stack2 是否为空，若为空，把 stack1 最后一个弹出给 stack2 （变成第一个进入栈2，最后出）；同时 stack1 若还有值则持续弹出给 stack2，原先在 stack1 的第一个最后出，变成最后一个进入 stack2（在栈顶）
  • 返回值：stack2 后进先出，从最后一个进入的弹出。
  • Special Case：若弹出的过程当中，stack2 只剩 {c}，stack1 有新的 d 被压入 ，则依照条件式判断 stack2 不为空时，此时弹出的仍是 stack2 里面有的值 {c}；若 stack为空，才会去 stack1 拿新压入的 {d} 做为弹出。

Python 代码

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

    def push(self, node):
        """
        压入
        """
        self.stack1.append(node)
            
        # write code here
    def pop(self):
        """
        弹出
        """
        if not self.stack2:
            while self.stack1:
                self.stack2.append(self.stack1.pop(-1))
        return self.stack2.pop(-1)