详解两个栈实现一个队列(python实现——经典面试题)

1、题目说明

      使用两个栈结构,实现一个队列功能,实现队列的添加元素和弹出元素。

      注意:

      队列特点是只能在队列尾部添加元素,在队列头部删除元素,先进先出(FIFO/LILO)

2、结题思路

       两个栈可以巧妙的结合,第一个栈用来添加元素,而在第二个栈中弹出元素。

       我们把所有添加元素的操作放在第一个栈中实现,当需要弹出元素时全部放在第二个栈中实现,当然,还会涉及有时候把第一个栈中的元素转移到第二个栈中已满足队列的逻辑。

3、代码实现

class QueueWithTwoStacks(object):
    def __init__(self):
        self._stack1 = []
        self._stack2 = []
    def appendTail(self,x):
        self._stack1.append(x)
    def deleteHead(self):
        if self._stack2:
            return self._stack2.pop()
        else:
            if self._stack1:
                while self._stack1:
                   self._stack2.append(self._stack1.pop())
                return self._stack2.pop()
            else:
                return None
    def getQueue(self):
        return self._stack1
q = QueueWithTwoStacks()
q.appendTail(1)
q.appendTail(2)
q.appendTail(3)
print(q.getQueue())
#q.deleteHead()
print(q.deleteHead())
print(q.deleteHead())
print(q.deleteHead())

猜你喜欢

转载自blog.csdn.net/su_bao/article/details/82346476