面试题9:用两个栈实现队列

这一题还是挺难的,有两个栈stackA、stackB,A是入栈的,B是出栈的,入栈时,直接进入A即可,出栈时,先判断是否有元素,

如果B没有元素,pop肯定报错,应该先将A中所有的元素压倒B里面,再pop最上面一个元素,如果B中有就直接pop出,就可以,

这是最优的思路,两个栈实现了先进后出,在一起又实现了队列的先进先出。

再次整理:

    1.首先init2个栈,一个做数据的输入栈A,一个做数据的输出栈B

    2.数据压入的时候,直接栈A.append节点

    3.数据输出的时候,需要考虑,这个时候的栈B是否为空,分两种情况,a.不为空,直接pop(); b.为空,那就在栈A不为空的情况下将栈A数据压入栈B,压完所有数据,就返回.pop()

 

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stackA = []
        self.stackB = []
    def push(self, node):
        # write code here
        self.stackA.append(node)
    def pop(self):
        if self.stackB != []:
            return self.stackB.pop()
        if self.stackB == []:
            while self.stackA:
                self.stackB.append(self.stackA.pop())
            return self.stackB.pop()

  

posted @ 2019-07-19 22:28  lililili——  阅读(225)  评论(0)    收藏  举报