《劍指offer》第5題:用兩個棧實現隊列

1 題目描述

  用兩個棧來實現一個隊列，完成隊列的Push和Pop操作。 隊列中的元素爲int類型。

2 分析及題解

  python中沒有棧，用list來代替棧。那麼定義兩個棧，一個用來push元素，一個用來pop元素

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

    def push(self, node):
        # write code here
        self.stack1.append(node)

    def pop(self):
        if len(self.stack1) == 0 and len(self.stack2) == 0:
            return None
        elif len(self.stack2) > 0:#stack2不爲空根據後入先出,棧頂元素出棧
            ele = self.stack2.pop()
            return ele
        else:
            while len(self.stack1) > 0:#stack2爲空，將stack1元素壓如stack2
                i = self.stack1.pop()
                self.stack2.append(i)
            ele = self.stack2.pop()#出棧
            return ele