溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
題目描述
用兩個隊列來實現一個棧，完成棧的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)
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
