题解 | #用两个栈实现队列#
用两个栈实现队列
https://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6
# -*- coding:utf-8 -*-
# from numpy import stack
class Solution:
def __init__(self):
self.stack1 = []
self.stack2 = []
def push(self, node):
self.stack1.append(node)
# 栈一用来暂存入队元素,栈二用来暂存出队元素
def pop(self):
if self.stack2:
temp = self.stack2.pop()
# 栈二有元素的情况下,栈顶元素即为出队元素
else:
while(self.stack1):
self.stack2.append(self.stack1.pop())
# 栈二为空时,将栈一的内容放到栈二,然后弹出栈顶元素,完成出队操作
temp = self.stack2.pop()
return temp

查看9道真题和解析