【leetcode】Python實現-225.用佇列實現棧
阿新 • • 發佈:2019-01-03
225.用佇列實現棧
描述
使用佇列實現棧的下列操作:
push(x) – 元素 x 入棧
pop() – 移除棧頂元素
top() – 獲取棧頂元素
empty() – 返回棧是否為空
注意
你只能使用佇列的基本操作– 也就是 push to back, peek/pop from front, size, 和 is empty 這些操作是合法的。
你所使用的語言也許不支援佇列。 你可以使用 list 或者 deque(雙端佇列)來模擬一個佇列 , 只要是標準的佇列操作即可。
你可以假設所有操作都是有效的(例如, 對一個空的棧不會呼叫 pop 或者 top 操作)。
依然使用列表實現了。
class MyStack:
def __init__(self):
"""
Initialize your data structure here.
"""
self.stack = []
def push(self, x):
"""
Push element x onto stack.
:type x: int
:rtype: void
"""
self.stack.append(x)
def pop(self):
"""
Removes the element on top of the stack and returns that element.
:rtype: int
"""
if self.stack == []:
return False
else:
return self.stack.pop()
def top(self):
"""
Get the top element.
:rtype: int
"""
if self.stack == []:
return False
else:
return self.stack[-1]
def empty(self):
"""
Returns whether the stack is empty.
:rtype: bool
"""
if self.stack == []:
return True
else:
return False
8.16更新,用兩個佇列實現棧。
思路:類似兩個棧實現佇列。queue1用來存放資料,每次入棧就插入queue1索引值為0的地方。queue2為出棧時的輔助佇列,將queue1[1:-1]的元素出隊然後入隊queue2(也就是queue1逆序壓入queue2),然後queue1僅剩隊首元素,這個時候出隊即出棧。
class MyStack(object):
def __init__(self):
"""
Initialize your data structure here.
"""
self.queue1 = []
self.queue2 = []
def push(self, x):
"""
Push element x onto stack.
:type x: int
:rtype: void
"""
self.queue1.insert(0, x)
def pop(self):
"""
Removes the element on top of the stack and returns that element.
:rtype: int
"""
for _ in range(len(self.queue1) - 1):
self.queue2.insert(0, self.queue1.pop())
res = self.queue1.pop()
self.queue1 = self.queue2
self.queue2 = []
return res
def top(self):
"""
Get the top element.
:rtype: int
"""
self.queue2 = self.queue1[:]
for i in range(len(self.queue1) - 1):
self.queue1.pop()
res = self.queue1[-1]
self.queue1 = self.queue2
self.queue2 = []
return res
def empty(self):
"""
Returns whether the stack is empty.
:rtype: bool
"""
if len(self.queue1) == 0:
return True
return False
# Your MyStack object will be instantiated and called as such:
# obj = MyStack()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.top()
# param_4 = obj.empty()