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(leetcode226)


下一篇:018 通过链表学Rust之实现pop front