两个栈实现一个队列

原创于 2018-12-22 00:32:02 发布 · 170 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #队列 #栈

数据结构专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种使用两个栈来模拟队列操作的方法。通过在第一个栈中压入元素，在需要弹出时将第一个栈的所有元素依次弹出并压入第二个栈，再从第二个栈中弹出元素，以此实现先进先出的队列特性。

class Two_stacks(object):
    def __init__(self):
        self._oneli=[]
        self._twoli=[]
    def entrance(self,item):
        """栈结构模拟进入队列"""
        self._oneli.append(item)
    def popup(self):
        """栈结构模拟弹出队列"""
        if self._twoli:
            # 如果第二个栈结构存在值就直接弹出
            return self._twoli.pop()
        else:
            # 如果不存在，将第一个栈循环从队尾弹出进入第二个栈的栈底
            if self._oneli:
                while self._oneli:
                    self._twoli.append(self._oneli.pop())
                return  self._twoli.pop()
            else:
            # 如果第一个栈内不存在值，直接返回空
                return None
if __name__ == '__main__':
    queue=Two_stacks()
    queue.entrance(1)
    queue.entrance(2)
    queue.entrance(3)
    queue.entrance(4)
    queue.entrance(5)
    print(queue.popup())
    print(queue.popup())
    print(queue.popup())
    print(queue.popup())
    print(queue.popup())