维持两个栈: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()