剑指offer Python版 - 用两个栈实现队列

xiaoxiao2021-02-28  41

题目描述

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

思路:定义两个栈stack1, stack2,当push元素时,将其插入进stack1,当pop元素时,若stack2为空,则将stack1中的元素依次弹出,并插入进stack2中,此时stack2中栈顶的元素是最早插入进stack1的,符合队列“先进先出”的特点,弹出stack2栈顶的元素。若stack2不为空,直接弹出栈顶元素。

# -*- coding:utf-8 -*- class Solution: def __init__(self): self.stack1 = [] self.stack2 = [] def push(self, node): # write code here self.stack1.append(node) def pop(self): # return xx if len(self.stack2) == 0: while self.stack1: self.stack2.append(self.stack1.pop()) return self.stack2.pop() return self.stack2.pop()

运行时间:23ms

占用内存:5624k

注:运行结果数据来源牛客网。

转载请注明原文地址: https://www.6miu.com/read-2625980.html

最新回复(0)