训练营day10|栈与队列|232.用栈实现队列、225. 用队列实现栈
232.用栈实现队列
要点
- python中栈可以用列表的append和pop函数实现
- 用两个栈实现队列的先进先出,即一个为输入栈,另一为输出栈,每次push都从输入栈压入数据,每次pop时,如果输出栈不为空,则从输出栈弹出数据,如果输出栈为空,则将输入栈全部数据按照后进先出的方式压入输出栈后再从输出栈弹出顶端数据。
- peek函数的实现相当于在pop实现后,将输出栈弹出的元素再压入输出栈。
代码
class MyQueue:
def __init__(self):
self.stack_in = []
self.stack_out = []
def push(self, x: int) -> None:
self.stack_in.append(x)
def pop(self) -> int:
if self.empty():
return None
if self.stack_out:
return self.stack_out.pop()
else:
for i in range(len(self.stack_in)):
self.stack_out.append(self.stack_in.pop())
return self.stack_out.pop()
def peek(self) -> int:
ans = self.pop()#直接使用pop函数
self.stack_out.append(ans)#因为pop函数弹出了,所以再添加回输出栈
return ans
def empty(self) -> bool:
return not (self.stack_in or self.stack_out)
225. 用队列实现栈
要点
- 可以用两个队列也可以用一个队列,思想一样
- python中需要用deque即双向队列来实现队列的先进先出操作,即popleft
- 时间和空间复杂度均为O(n)
代码
class MyStack:
def __init__(self):
self.que = deque()
def push(self, x: int) -> None:
self.que.append(x)
def pop(self) -> int:
if self.empty():
return None
for i in range(len(self.que) - 1):
self.que.append(self.que.popleft())
return self.que.popleft()
def top(self) -> int:
if self.empty():
return None
return self.que[-1]
def empty(self) -> bool:
return not self.que