题目描述:
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
解题思路:
1.栈是先进后出的,队列是先进先出的,如下图。
2.把新添加进来的节点都添加到栈1上,然后栈1上的节点添加到栈2上,再将栈2上的节点输出,就是实现了队列的输出方式。输出pop()函数具体可以分为以下三种情况。
(1)当栈2非空时,直接将栈2的最后一个节点输出(由于栈的特点时先进后出,后进先出的,故先输出的是最后一个节点)
(2)当栈1为空,栈2非空,将栈1内的节点从后往前逐个加入到栈2中,最后再输出删除掉栈2中最后一个节点
(3)当栈1、栈2都为空时,说明队列已经完成输出了
python代码如下:
# -*- coding:utf-8 -*-
class Solution:
def __init__(self):
self.stack1 = []
self.stack2 = []
def push(self, node):
# write code here
self.stack1.append(node)
def pop(self):
if self.stack2:##栈2非空,直接删掉栈2的最后一个节点
return self.stack2.pop()
elif self.stack1:##如果栈1,栈2都为空,算法结束
"""
栈1为空,栈2非空,将栈1内的节点从后往前逐个加入到栈2中,
最后再删掉栈2中最后一个节点
"""
while self.stack1:
self.stack2.append(self.stack1.pop())
return self.stack2.pop()
else:
return
ps:
列表 List pop(obj)函数用于移除列表中的一个元素(obj为可选参数,可以省略,默认最后一个元素),并且返回该元素的值。