乐扣挑战：栈实现队列

2024-01-26 23:46:44

使用栈实现队列的原理

栈是一种先进后出的数据结构，而队列是一种先入先出的数据结构。为了使用栈实现队列，我们需要利用栈的特性来模拟队列的行为。

具体来说，我们可以使用两个栈来实现队列。一个栈用于存储入队的元素，另一个栈用于存储出队的元素。当我们入队一个元素时，我们将该元素压入第一个栈。当我们出队一个元素时，我们将第一个栈中的所有元素依次弹出并压入第二个栈，然后从第二个栈中弹出最顶部的元素作为出队的元素。

代码示例

class Queue:
    def __init__(self):
        self.in_stack = []
        self.out_stack = []

    def enqueue(self, item):
        self.in_stack.append(item)

    def dequeue(self):
        if not self.out_stack:
            while self.in_stack:
                self.out_stack.append(self.in_stack.pop())
        return self.out_stack.pop()

    def peek(self):
        if not self.out_stack:
            while self.in_stack:
                self.out_stack.append(self.in_stack.pop())
        return self.out_stack[-1]

    def is_empty(self):
        return not self.in_stack and not self.out_stack


# 测试代码
queue = Queue()
queue.enqueue(1)
queue.enqueue(2)
queue.enqueue(3)

print(queue.dequeue())  # 输出：1
print(queue.peek())  # 输出：2
print(queue.is_empty())  # 输出：False