mongona

class Solution(object):
    """
    两个栈实现一个队列
    """
    def __init__(self):
        # -----
        self.stack1 = []
        self.stack2 = []

    def push(self, element):
        self.stack1.append(element)

    def pop(self):
        if self.stack2 == [] and self.stack1 == []:
            raise IndexError("index out of range, stack is empty now!")

        for val in range(len(self.stack1)):
            self.stack2.append(self.stack1.pop(0))
        return self.stack2.pop(0)

life is a boat

2

2

Author：Sato Please give credit to the original author when you use it elsewhere：https://www.mongona.com/blog/10

Pre Page
两个队列实现一个栈

Next Page
hello lua

Tags

python 数据结构

富强，民主，文明，和谐，自由，平等，公正，法治，爱国，敬业，诚信，友善。

微信 支付宝

赏