更正 0232.用栈实现队列.md python代码

添加限制条件,仅在栈不为空的情况下才允许后续pop()和peek()的操作
This commit is contained in:
Eyjan_Huang
2021-08-20 03:29:26 +08:00
committed by GitHub
parent 8b94306cc6
commit ca576c5bd0

View File

@ -304,6 +304,9 @@ class MyQueue:
2. 如果out没有元素就把in里面的元素除了第一个依次pop后装进out里面 2. 如果out没有元素就把in里面的元素除了第一个依次pop后装进out里面
3. 直接把in剩下的元素pop出来就是queue头部的 3. 直接把in剩下的元素pop出来就是queue头部的
""" """
if self.empty:
return None
if self.stack_out: if self.stack_out:
return self.stack_out.pop() return self.stack_out.pop()
else: else:
@ -317,6 +320,9 @@ class MyQueue:
1. 查out有没有元素有就把最上面的返回 1. 查out有没有元素有就把最上面的返回
2. 如果out没有元素就把in最下面的返回 2. 如果out没有元素就把in最下面的返回
""" """
if self.empty:
return None
if self.stack_out: if self.stack_out:
return self.stack_out[-1] return self.stack_out[-1]
else: else: