今天的每日一题是用栈实现队列,是一个简单题。先来看一下题目描述吧。
最开始看到这个题我想到的是用一个栈来模拟队列,但是python里面没有栈这个数据结构,只能用列表来模拟栈。但是实现完之后提交虽然通过了,我回来再仔细一看感觉有些不对,题目要求只能用栈的操作,但是我好像还用到了insert函数,我是直接把一个列表当做一个栈来使用了,这样虽然能通过测试,但是是不符合题目要求的。于是看了一下官方题解,原来可以使用两个栈来模拟队列。一个栈作为输入,一个栈作为输出,当输出的栈元素全部都弹出去后,就把输入的栈的与元素都放到输出栈,这样就实现了一个队列。而且除了将输入栈元素放入输出栈,其他操作都是O(1)的时间复杂度。
class MyQueue:
def __init__(self):
"""
Initialize your data structure here.
"""
self.instack = []
self.outstack = []
def push(self, x: int) -> None:
"""
Push element x to the back of queue.
"""
self.instack.append(x)
def pop(self) -> int:
"""
Removes the element from in front of queue and returns that element.
"""
if len(self.outstack)==0:
self.in_to_out()
n = len(self.outstack)
return self.outstack.pop(n-1)
def peek(self) -> int:
"""
Get the front element.
"""
if len(self.outstack)==0:
self.in_to_out()
n = len(self.outstack)
return self.outstack[n-1]
def empty(self) -> bool:
"""
Returns whether the queue is empty.
"""
if len(self.instack)==0 and len(self.outstack)==0:
return True
else:
return False
def in_to_out(self) -> None:
while(self.instack):
n = len(self.instack)
self.outstack.append(self.instack[n-1])
del self.instack[n-1]
# 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()
官方题解没有给出python的参考代码,这个是我自己写的,有些混乱,将就着可以看一下