维持两个栈:stack1负责插入,stack2负责删除

  • 当进队列时,直接插入stack1;

  • 当出队列时,

    若stack2为空,则将stack1的元素依次弹出到stack2

    若stack2不为空,则弹出stack2栈顶元素

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack1 = []  # 负责插入
        self.stack2 = []  # 负责删除
    def push(self, node):
        self.stack1.append(node)
        
    def pop(self):
        # stack2为空时,将stack1元素弹到stack2
        if not self.stack2:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
        # 弹出stack2栈顶元素
        return self.stack2.pop()