栈与队列part01
今日内容: 栈与队列理论基础、用栈实现队列、用队列实现栈
代码随想录链接:代码随想录 (programmercarl.com)
菜菜刷题第一天,为了赶上大部队进度从day10开始
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 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
思路:
第一次刷题也是绞尽脑汁啊,只知道队是先进先出、栈是先进后出,其他一脸懵逼。在看了视频的思路后发现,队列的逻辑其实就是栈的颠倒,所以就是把栈里的元素扔到另一个栈里然后再出,逐渐有了思路。于是开始手搓py代码。
我的代码:
其中用到 pop()
函数,pop()
函数是Python中的一个列表方法,用于移除列表中的一个元素(默认最后一个元素),并且返回该元素的值。一般 pop() 是最后一个元素,而 pop(0) 是第一个元素。于是搓出了我的垃圾代码。
class MyQueue:
def __init__(self):
self.stack1 = []
self.stack2 = []
def push(self, x: int) -> None:
self.stack1.append(x)
def pop(self) -> int:
if not self.stack2:
while self.stack1:
self.stack2.append(self.stack1.pop())
a = self.stack2.pop()
if not self.stack1:
while self.stack2:
self.stack1.append(self.stack2.pop())
return a
def peek(self) -> int:
if not self.stack2:
while self.stack1:
self.stack2.append(self.stack1.pop())
b = self.stack2[-1]
if not self.stack1:
while self.stack2:
self.stack1.append(self.stack2.pop())
return b
def empty(self) -> bool:
if len(self.stack1)==0:
return True
else:
return False
答案:
经过比较感觉答案是高级了许多:
1、写完的pop函数可以直接在peek函数里调用
2、stack1中的元素扔到另一个中后可以不用再扔回来了
3、在函数进行前先判断是否是empty,提高效率
class MyQueue:
def __init__(self):
"""
in主要负责push,out主要负责pop
"""
self.stack_in = []
self.stack_out = []
def push(self, x: int) -> None:
"""
有新元素进来,就往in里面push
"""
self.stack_in.append(x)
def pop(self) -> int:
"""
Removes the element from in front of queue and returns that element.
"""
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:
"""
Get the front element.
"""
ans = self.pop()
self.stack_out.append(ans)
return ans
def empty(self) -> bool:
"""
只要in或者out有元素,说明队列不为空
"""
return not (self.stack_in or self.stack_out)
225. 用队列实现栈
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push
、top
、pop
和 empty
)。
实现 MyStack
类:
void push(int x)
将元素 x 压入栈顶。int pop()
移除并返回栈顶元素。int top()
返回栈顶元素。boolean empty()
如果栈是空的,返回true
;否则,返回false
。
说明:
- 你只能使用队列的基本操作 —— 也就是
push to back
、peek/pop from front
、size
和is empty
这些操作。 - 你所使用的语言也许不支持队列。 你可以使用 list (列表)或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
思路:
已经做了一题后,第二题略清晰了些。仔细思考,不就是将队列中最后进的保留,然后将前面的所有元素先转移出来,然后再将最后的输出嘛。写的过程中发现,咦,好像直接用一个队列将前面出来的转移到尾部不就好了吗,恍然大悟!于是又看了视频,思路如出一辙。
我的代码:
class MyStack:
def __init__(self):
self.stack1 = []
def push(self, x: int) -> None:
self.stack1.append(x)
def pop(self) -> int:
if self.empty():
return None
size = len(self.stack1)
for i in range(size-1):
self.stack1.append(self.stack1.pop(0))
a = self.stack1.pop(0)
return a
def top(self) -> int:
b = self.pop()
self.stack1.append(b)
return b
def empty(self) -> bool:
if len(self.stack1) == 0:
return True
else:
return False
答案:
在提函数的过程中我用的是 pop(0)
,看完答案发现原来也可以用 popleft(),学到了。
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]
# 写法二:
if self.empty():
return None
for i in range(len(self.que)-1):
self.que.append(self.que.popleft())
temp = self.que.popleft()
self.que.append(temp)
return temp
def empty(self) -> bool:
return not self.que