劍指offer:兩個隊列模擬一個棧
阿新 • • 發佈:2019-04-17
[] return 作用 clas n+1 假設 實現 node urn 題目描述
用兩個隊列來實現一個棧,完成棧的Push和Pop操作。 隊列中的元素為int類型。
用兩個隊列來實現一個棧,完成棧的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:兩個隊列模擬一個棧