队列和栈第一天
队列和栈的基础知识
卡哥链接
注意:
- 队列和栈本身不是容器;
- 栈提供push 和 pop 等等接口,所有元素必须符合先进后出规则,所以栈不提供走访功能,也不提供迭代器(iterator)。 不像是set 或者map 提供迭代器iterator来遍历所有元素;
- 队列所有元素必须符合先进先出规则;
232.用栈实现队列
题目
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):
实现 MyQueue 类:
void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空,返回 true ;否则,返回 false
说明:
只能 使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
这是一个模拟题,不涉及具体的算法,考察的就是对栈和队列的掌握程度。使用 list 来模拟一个栈,只要是标准的栈操作即可。
self.pop()可以被复用,学习这种思想。
class MyQueue:
def __init__(self):
self.instack = []
self.outstack = []
def push(self, x: int) -> None:
self.instack.append(x)
def pop(self) -> int:
if self.empty():
return None
if len(self.outstack)==0:
# 如果instack有值,就把它全部加到outstack中
# python list的pop会弹出最后一个元素
while self.instack:
self.outstack.append(self.instack.pop())
return self.outstack.pop()
def peek(self) -> int:
res = self.pop()
# 因为只是返回第一个元素,但是直接pop会把他删掉,就需要恢复
self.outstack.append(res)
# 或者可以把元素加入到outstack后用list返回最后一个元素
return res
def empty(self) -> bool:
# 两个栈都为空,则队列为空
return len(self.instack)==0 and len(self.outstack)==0
时间复杂度: push和empty为O(1), pop和peek为O(n)
空间复杂度: O(n)
225. 用队列实现栈
题目
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。
队列模拟栈,其实一个队列就够了
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
时间复杂度: pop为O(n),其他为O(1)
空间复杂度: O(n)