Python教程

python两个栈实现一个队列

本文主要是介绍python两个栈实现一个队列,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
#两个栈实现一个队列:只使用一个栈 stack1 当作队列,另一个栈 stack2 用来辅助操作。
#要想将新加入的元素出现栈底,需要先将 stack1 的元素转移到 stack2,将元素入栈 stack1,最后将 stack2 的元素全部回到 stack1。
class cQueue:
    def __init__(self):
        self.A,self.B=[],[]
    def appendTail(self,value):
        self.A.append(value)
    def deleteHead(self):
        if self.B:return self.B.pop()
        if not self.A:return -1
        while self.A:
            self.B.append(self.A.pop())
        return self.B.pop()

if __name__=='__main__':
    cqueue=cQueue()
    a=[3,5,6,7,4]
    for i in a:
        cqueue.appendTail(i)
    print(cqueue.A)
    cqueue.deleteHead()
    print(cqueue.B)
    print(cqueue.B.pop())
 #时间复杂度O(1),空间复杂度为O(n)
这篇关于python两个栈实现一个队列的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!