题解 | 两个队列实现栈

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
#
# @param order string字符串一维数组
# @return string字符串一维数组
#
class Solution:
    def __init__(self):
        self.s = []

    def push(self, c: str):
        self.s.append(c)

    def top(self):
        return self.s[-1]

    def pop(self):
        return self.s.pop()

    def empty(self):
        return True if len(self.s) == 0 else False

    def simulate(self, order: List[str]) -> List[str]:
        # write code here
        ans = []
        for p in order:
            if p == "TOP":
                ans.append(self.top())
            elif p == "POP":
                ans.append(self.pop())
            elif p == "MTY":
                ans.append(self.empty())
            else:  #'PSH'
                self.push(p[3:])
            print(ans)
        return ans
#注意要写成类和子方法的模式