使用栈实现队列的下列操作:
- push(x) – 将一个元素放入队列的尾部。
- pop() – 从队列首部移除元素。
- peek() – 返回队列首部的元素。
- empty() – 返回队列是否为空。
说明:
你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)。——>判断代码健壮性
思路
这道题考察的就是对栈和队列的掌握程度。
使用栈来模式队列的行为,如果仅仅用一个栈,是一定不行的,所以需要两个栈一个输入栈,一个输出栈,这里要注意输入栈和输出栈的关系。
在push数据的时候,只要数据放进输入栈就好,但在pop的时候,操作就复杂一些,输出栈如果为空,就把进栈数据全部导入进来(注意是全部导入),再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据就可以了。
最后如何判断队列为空呢?如果进栈和出栈都为空的话,说明模拟的队列为空了。
代码
注意一些细节:文章来源:https://www.toymoban.com/news/detail-723043.html
- 判断栈是否为空,才能够进行有效操作
- 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 None
if 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 ans
def 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文章来源地址https://www.toymoban.com/news/detail-723043.html
到了这里,关于(※)力扣刷题-栈和队列-用栈实现队列的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!