File tree Expand file tree Collapse file tree 1 file changed +8
-15
lines changed Expand file tree Collapse file tree 1 file changed +8
-15
lines changed Original file line number Diff line number Diff line change @@ -205,33 +205,26 @@ class MyQueue:
205
205
206
206
def pop (self ) -> int :
207
207
"""
208
- 1. 检查如果out里面元素,则直接pop
209
- 2. 如果out没有元素,就把in里面的元素(除了第一个)依次pop后装进out里面
210
- 3. 直接把in剩下的元素pop出来,就是queue头部的
208
+ Removes the element from in front of queue and returns that element.
211
209
"""
212
- if self .empty:
210
+ if self .empty() :
213
211
return None
214
212
215
213
if self .stack_out:
216
214
return self .stack_out.pop()
217
215
else :
218
- for i in range (1 , len (self .stack_in)):
216
+ for i in range (len (self .stack_in)):
219
217
self .stack_out.append(self .stack_in.pop())
220
- return self .stack_in .pop()
218
+ return self .stack_out .pop()
221
219
222
220
223
221
def peek (self ) -> int :
224
222
"""
225
- 1. 查out有没有元素,有就把最上面的返回
226
- 2. 如果out没有元素,就把in最下面的返回
223
+ Get the front element.
227
224
"""
228
- if self .empty:
229
- return None
230
-
231
- if self .stack_out:
232
- return self .stack_out[- 1 ]
233
- else :
234
- return self .stack_in[0 ]
225
+ ans = self .pop()
226
+ self .stack_out.append(ans)
227
+ return ans
235
228
236
229
237
230
def empty (self ) -> bool :
You can’t perform that action at this time.
0 commit comments