两个队列实现一个栈

2022-09-07 11:28:04 浏览数 (1)

代码语言:javascript复制
class Solution(object):
    """
    两个队列实现一个栈
    两个队列来回倒
    """
    def __init__(self):
        """
        初始化两个队列
        """
        self.queue1 = []
        self.queue2 = []

    def push(self, element):
        # 入栈时 选择一个空的队列
        if self.queue1 == []:
            self.queue2.append(element)
        else:
            self.queue1.append(element)

    def pop(self):
        # 栈为空时 抛异常
        if self.queue1 == [] and self.queue2 == []:
            raise IndexError("index out of range,  stack is empty") 

        elif self.queue2 == []:
            # queue2 为空的时候,把queue1的除了最后一个数全给添加到queue2
            for val in range(len(self.queue1) - 1):
                self.queue2.append(self.queue1.pop(0))
            # 弹出 queue1的元素  只有一个元素
            return self.queue1.pop()

        else:
            # queue1 为空的时候,把queue2的除了最后一个数全给添加到queue1
            for val in range(len(self.queue2) -1):
                self.queue1.append(self.queue2.pop(0))
            # 弹出 queue2的元素  只有一个元素
            return self.queue2.pop()

life is short

0 人点赞