[5]オファーキューが実装2つのスタックでのpythonを獲得します

タイトル説明

キューを達成するための2つのスタック、完了キューのプッシュとポップ操作。キューの要素がint型。

アイデア：

• スタックはキューとして使用されています
• スタックBが空である場合、スタックのためのBキュー、すべてのB、Bスタックを積層するスタックと、スタックのスタック（すなわち、キュー）

コード：

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stackA = []
        self.stackB = []

    def push(self, node):
        # write code here
        self.stackA.append(node)

    def pop(self):
        # return xx
        if self.stackB:
            return self.stackB.pop()

        elif not self.stackA:
            return None
        else:

            # while 循环
            while self.stackA:
                #
                self.stackB.append(self.stackA.pop())  # 默认弹出列表的最后一个元素
            # print("===",self.stackB)  # [4, 3, 2, 1]
            return self.stackB.pop()

    def getQueue(self):
        print("stackA", self.stackA)
        print("stackB", self.stackB)


# 我大Python的内建数据结构太强大，可以用list直接实现栈，简单快捷。
sta = Solution()
sta.push(1)
sta.getQueue()
sta.push(2)
sta.getQueue()
sta.push(3)
sta.getQueue()
sta.push(4)
sta.getQueue()  # stackA [1, 2, 3, 4] stackB []

print(sta.pop())  # 1
sta.getQueue()  # stackA [] stackB [4, 3, 2]
print(sta.pop())  # 2
sta.getQueue()  # stackA [] stackB [4, 3]
print(sta.pop())  # 3
sta.getQueue()  # stackA [] stackB [4]
print(sta.pop())  # 4
sta.getQueue()  # stackA [] stackB []