python 队列实现一个栈_python练习之两个栈实现队列

用两个栈实现队列，实现入队和出队方法

代码：

class Queue(object):

def __init__(self):

self.stack1 = []

self.stack2 = []

#入队

def enqueue(self,element):

if self.stack1:

self.stack1.append(element)

else:

while self.stack2:

self.stack1.append(self.stack2.pop())

self.stack1.append(element)

#出队

def dequeue (self):

if self.stack2:

return self.stack2.pop()

elif self.stack1:

while self.stack1:

self.stack2.append(self.stack1.pop())

return self.stack2.pop()

else:

return None

if __name__=='__main__':

queue = Queue()

queue.enqueue('9')

queue.enqueue('10')

queue.enqueue('6')

queue.enqueue('2')

print queue.dequeue()

print queue.dequeue()

print queue.dequeue()

print queue.dequeue()

运行结果为：

9

10

6

2

思路：

入队时，先判断s1是否为空，如不为空，说明所有元素都在s1，此时将入队元素直接压入s1；如为空，要将s2的元素逐个“倒回”s1，再压入入队元素。

出队时，先判断s2是否为空，如不为空，直接弹出s2的顶元素并出队；如为空，将s1的元素逐个“倒入”s2，把最后一个元素弹出并出队。