Python算法——用两个栈模拟队列操作

最新推荐文章于 2024-06-10 22:03:48 发布

Aamax

最新推荐文章于 2024-06-10 22:03:48 发布

阅读量350

点赞数 2

分类专栏：算法

本文链接：https://blog.csdn.net/xjtuse123/article/details/98472270

版权

该博客探讨如何使用两个栈A和B来模拟队列Q，详细阐述了入队（栈A）和出队（栈B）的操作流程。在出队时，若栈B非空直接弹出，否则将栈A所有元素转移至栈B再弹出。文章还分析了性能，指出入队操作时间复杂度为O(1)，出队操作时间复杂度在平均情况下也是O(1)。

摘要由CSDN通过智能技术生成

要求：

用两个栈模拟队列，假设使用栈A与栈B模拟队列Q，A为插入栈，B为弹出栈，以实现队列Q。

分析：

假设A和B都为空，可以认为A提供入队功能，栈B提供出队功能。要入队列，入栈A即可，而出队列则需要分两种情况考虑：

(1)如果栈B不为空，则直接弹出栈B数据

(2)如果栈B为空，则依次弹出栈A的数据，放入栈B中，再弹出栈B的数据。

# -*- coding:utf-8 -*-
class Stack():
    #模拟栈
    def __init__(self):
        self.items = []
    #判断栈是否为空
    def empty(self):
        return len(self.items)==0
    #返回栈的大小
    def size(self):
        return len(self.items)
    #返回栈顶元素
    def peek(self):
        if not self.empty():
            return self.items[len(self.items)-1]
        else:
            return None
    #出栈
    def pop(self):
        if len(self.items)>0:
            return self.items.pop()
        else:
            print("栈已经为空！")
            return None
    #入栈
    def push(self,item):