1. 程式人生 > >劍指offer:兩個隊列模擬一個棧

劍指offer:兩個隊列模擬一個棧

[] return 作用 clas n+1 假設 實現 node urn

題目描述
用兩個隊列來實現一個棧,完成棧的Push和Pop操作。 隊列中的元素為int類型。

實現方式其實和兩個棧模擬一個隊列相似,但是區別在於這兩個隊列的作用和那兩個棧的作用不一樣。

class Solution:
    """
    用兩個隊列模擬一個棧,如果兩個隊列的容量分別為M和N,其中M > N,那麽模擬得到的棧的容量是N+1
    因為假設先把queue1塞進N+2個,此時將元素出棧,則需要先將queue1的N+1個元素出隊後壓入queue2,
    由於queue2的容量只有N,入隊失敗。因此,最大容量是N+1
    """
    def __init__(self):
        # queue1和queue2都可以作為入棧的隊列,也可以作為出棧的輔助隊列,作用一樣。不像用棧模擬
        # 隊列時一個stack是作為入隊的棧,另一個stack作為出隊的棧。
        self.queue1 = []
        self.queue2 = []

    def push(self, node):
        # 入棧的時候往非空的隊列添加
        if self.queue1:
            self.queue1.append(node)
        else:
            self.queue2.append(node)

    def pop(self):
        if not self.queue1 and not self.queue2:
            return None
        # 出棧的時候需要先將非空隊列中的前N-1個元素順序壓入另一個隊列,然後是彈出最後一個元素
        if self.queue1:
            while len(self.queue1) > 1:
                self.queue2.append(self.queue1.pop(0))
            return self.queue1.pop(0)
        else:
            while len(self.queue2) > 1:
                self.queue1.append(self.queue2.pop(0))
            return self.queue2.pop(0)

劍指offer:兩個隊列模擬一個棧