232.用栈实现队列(python)

文章描述了如何使用两个栈实现一个支持push、pop、peek和empty操作的MyQueue类,以保持先入先出的队列特性。作者详细解释了如何在stack1进行入队操作,以及在stack2进行出队和查看队头操作的逻辑。
摘要由CSDN通过智能技术生成

题目要求:

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(pushpoppeekempty):

实现 MyQueue 类:

  • void push(int x) 将元素 x 推到队列的末尾
  • int pop() 从队列的开头移除并返回元素
  • int peek() 返回队列开头的元素
  • boolean empty() 如果队列为空,返回 true ;否则,返回 false

说明:

  • 你 只能 使用标准的栈操作 —— 也就是只有 push to toppeek/pop from topsize, 和 is empty 操作是合法的。
  • 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。

示例 1:

输入:
["MyQueue", "push", "push", "peek", "pop", "empty"]
[[], [1], [2], [], [], []]
输出:
[null, null, null, 1, 1, false]

解释:
MyQueue myQueue = new MyQueue();
myQueue.push(1); // queue is: [1]
myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue)
myQueue.peek(); // return 1
myQueue.pop(); // return 1, queue is [2]
myQueue.empty(); // return false

要使用两个栈实现先入先出队列,可以利用两个栈来模拟队列的入队和出队操作。

class MyQueue:

    def __init__(self):
        # 用于入队
        self.stack1 = []
        # 用于出队
        self.stack2 = []

    def push(self, x: int) -> None:
        # 将元素推到队列的末尾,即入栈到stack1
        self.stack1.append(x)

    def pop(self) -> int:
        # 如果stack2为空,将stack1中的元素依次弹出并压入stack2
        if not self.stack2:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
        # 弹出stack2的栈顶元素,即队列的开头元素
        return self.stack2.pop()

    def peek(self) -> int:
        # 如果stack2为空,将stack1中的元素依次弹出并压入stack2
        if not self.stack2:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
        # 返回stack2的栈顶元素,即队列的开头元素,但不弹出
        return self.stack2[-1]

    def empty(self) -> bool:
        # 队列为空的条件是两个栈都为空
        return not self.stack1 and not self.stack2

使用两个栈模拟队列的基本操作,包括入队(push)、出队(pop)、查看队头元素(peek)和判断队列是否为空(empty)。

  1. 初始化:

    • __init__ 方法中,创建了两个栈,stack1 用于入队操作,stack2 用于出队操作。
  2. 入队操作(push):

    • push 方法将元素 x 入栈到 stack1 中,实现了队列的入队操作。
  3. 出队操作(pop):

    • pop 方法首先检查 stack2 是否为空。
    • 如果 stack2 不为空,直接从 stack2 出栈一个元素,即模拟队列的出队操作。
    • 如果 stack2 为空,将 stack1 中的元素逐个出栈并压入 stack2 中,然后从 stack2 出栈一个元素。这样确保了元素按照入队的顺序出队。
  4. 查看队头元素(peek):

    • peek 方法同样检查 stack2 是否为空。
    • 如果 stack2 不为空,返回 stack2 的栈顶元素,即队列的开头元素,但不出队。
    • 如果 stack2 为空,将 stack1 中的元素逐个出栈并压入 stack2 中,然后返回 stack2 的栈顶元素。
  5. 判断队列是否为空(empty):

    • empty 方法返回一个布尔值,判断队列是否为空。队列为空的条件是两个栈都为空。

这个实现保持了队列的先入先出的顺序。在出队或查看队头元素时,如果 stack2 不为空,直接从 stack2 中操作,保持了入队顺序;如果 stack2 为空,将 stack1 中的元素转移到 stack2 中,实现了将元素按照入队顺序出队。

复杂度分析:

  1. 入队操作 (push):

    • 入队操作只涉及将元素压入 stack1,这是一个常数时间的操作,时间复杂度为 O(1)。
  2. 出队操作 (pop):

    • 出队操作在最坏的情况下,需要将 stack1 中的所有元素逐个弹出并压入 stack2,然后再从 stack2 中弹出一个元素。这个过程的时间复杂度是 O(n),其中 n 是队列中的元素数量。但是由于每个元素只会被移动一次,对于整个序列的操作,平均下来每个元素仍然是常数时间内完成的。
    • 在均摊意义下,出队操作的时间复杂度可以视为 O(1)。
  3. 查看队头元素 (peek):

    • 类似于出队操作,查看队头元素在最坏情况下也需要 O(n) 的时间,因为需要将元素从 stack1 移到 stack2
    • 在均摊意义下,查看队头元素的时间复杂度也可以视为 O(1)。
  4. 判断队列是否为空 (empty):

    • 判断两个栈是否都为空是一个常数时间的操作,时间复杂度为 O(1)。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值