class twoStackOneQueue(object):
def __init__(self):
\\初始化栈
self.stack1=[]
self.stack2=[]
def push(self,item):
\\第一个栈追加元素
self.stack1.append(item)
def pop(self):
\\判断第二个栈是否有元素,有元素则弹出
if self.stack2:
self.stack2.pop()
\\没有执行
else:
\\判断栈1是否有元素
if self.stack1:
\\如果栈一有元素,则向栈2追加
while self.stack1:
self.stack2.append(self.stack1.pop())
\\出栈
return self.stack2.pop()
else:
return None
queue=twoStackOneQueue()
queue.push(1)
queue.push(2)
queue.push(3)
queue.push(4)
queue.push(5)
print(queue.pop())
queue.push(6)
print(queue.pop())