題目描述
用兩個隊列來實現(xiàn)一個棧,完成棧的Push和Pop操作。 隊列中的元素為int類型。
實現(xiàn)方式其實和兩個棧模擬一個隊列相似,但是區(qū)別在于這兩個隊列的作用和那兩個棧的作用不一樣。
class Solution:
"""
用兩個隊列模擬一個棧,如果兩個隊列的容量分別為M和N,其中M > N,那么模擬得到的棧的容量是N+1
因為假設(shè)先把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)
另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)cdcxhl.cn,海內(nèi)外云服務(wù)器15元起步,三天無理由+7*72小時售后在線,公司持有idc許可證,提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國服務(wù)器、虛擬主機、免備案服務(wù)器”等云主機租用服務(wù)以及企業(yè)上云的綜合解決方案,具有“安全穩(wěn)定、簡單易用、服務(wù)可用性高、性價比高”等特點與優(yōu)勢,專為企業(yè)上云打造定制,能夠滿足用戶豐富、多元化的應(yīng)用場景需求。