diff --git a/problems/0232.用栈实现队列.md b/problems/0232.用栈实现队列.md index 46d884d3..27a3de4e 100644 --- a/problems/0232.用栈实现队列.md +++ b/problems/0232.用栈实现队列.md @@ -304,6 +304,9 @@ class MyQueue: 2. 如果out没有元素,就把in里面的元素(除了第一个)依次pop后装进out里面 3. 直接把in剩下的元素pop出来,就是queue头部的 """ + if self.empty: + return None + if self.stack_out: return self.stack_out.pop() else: @@ -317,6 +320,9 @@ class MyQueue: 1. 查out有没有元素,有就把最上面的返回 2. 如果out没有元素,就把in最下面的返回 """ + if self.empty: + return None + if self.stack_out: return self.stack_out[-1] else: