使用栈实现队列的下列操作:
- push(x) – 将一个元素放入队列的尾部。
- pop() – 从队列首部移除元素。
- peek() – 返回队列首部的元素。
- empty() – 返回队列是否为空。
说明:
你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)。——>判断代码健壮性
思路
这道题考察的就是对栈和队列的掌握程度。
使用栈来模式队列的行为,如果仅仅用一个栈,是一定不行的,所以需要两个栈一个输入栈,一个输出栈,这里要注意输入栈和输出栈的关系。
在push数据的时候,只要数据放进输入栈就好,但在pop的时候,操作就复杂一些,输出栈如果为空,就把进栈数据全部导入进来(注意是全部导入),再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据就可以了。
最后如何判断队列为空呢?如果进栈和出栈都为空的话,说明模拟的队列为空了。
代码
注意一些细节:
- 判断栈是否为空,才能够进行有效操作
- pop时候的复杂性
- peek复用pop后需要把元素加回去
class MyQueue(object):def __init__(self):"""in 主要负责push out主要负责pop(用栈实现队列 必然是要用到两个栈)"""self.stack_in = [] # python中没有栈这个数据结构 用列表代替self.stack_out = []def push(self, x):""":type x: int:rtype: None"""# 新元素进来就往 in 中加入 # 在push数据的时候,只要数据放进输入栈就好self.stack_in.append(x)def pop(self):""":rtype: int"""# 在pop的时候,操作就复杂一些,输出栈如果为空,就把进栈数据全部导入进来(注意是全部导入),# 再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据就可以了。# 从队列首部移除元素if self.empty():return Noneif self.stack_out: # stack_out中有元素return self.stack_out.pop() # 这里用return 既可以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):""":rtype: int"""ans = self.pop() # 代码 函数复用self.stack_out.append(ans) # 因为只是返回元素 所以采用pop删除之后(代码复用)还需要再加回去return ansdef empty(self):""":rtype: bool"""return not (self.stack_in or self.stack_out)# Your MyQueue object will be instantiated and called as such:
# obj = MyQueue()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.peek()
# param_4 = obj.empty()
参考:https://www.programmercarl.com/0232.%E7%94%A8%E6%A0%88%E5%AE%9E%E7%8E%B0%E9%98%9F%E5%88%97.html