题目要求:
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push
、pop
、peek
、empty
):
实现 MyQueue
类:
void push(int x)
将元素 x 推到队列的末尾int pop()
从队列的开头移除并返回元素int peek()
返回队列开头的元素boolean empty()
如果队列为空,返回true
;否则,返回false
说明:
- 你 只能 使用标准的栈操作 —— 也就是只有
push to top
,peek/pop from top
,size
, 和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
)。
-
初始化:
- 在
__init__
方法中,创建了两个栈,stack1
用于入队操作,stack2
用于出队操作。
- 在
-
入队操作(push):
push
方法将元素x
入栈到stack1
中,实现了队列的入队操作。
-
出队操作(pop):
pop
方法首先检查stack2
是否为空。- 如果
stack2
不为空,直接从stack2
出栈一个元素,即模拟队列的出队操作。 - 如果
stack2
为空,将stack1
中的元素逐个出栈并压入stack2
中,然后从stack2
出栈一个元素。这样确保了元素按照入队的顺序出队。
-
查看队头元素(peek):
peek
方法同样检查stack2
是否为空。- 如果
stack2
不为空,返回stack2
的栈顶元素,即队列的开头元素,但不出队。 - 如果
stack2
为空,将stack1
中的元素逐个出栈并压入stack2
中,然后返回stack2
的栈顶元素。
-
判断队列是否为空(empty):
empty
方法返回一个布尔值,判断队列是否为空。队列为空的条件是两个栈都为空。
这个实现保持了队列的先入先出的顺序。在出队或查看队头元素时,如果 stack2
不为空,直接从 stack2
中操作,保持了入队顺序;如果 stack2
为空,将 stack1
中的元素转移到 stack2
中,实现了将元素按照入队顺序出队。
、
复杂度分析:
-
入队操作 (
push
):- 入队操作只涉及将元素压入
stack1
,这是一个常数时间的操作,时间复杂度为 O(1)。
- 入队操作只涉及将元素压入
-
出队操作 (
pop
):- 出队操作在最坏的情况下,需要将
stack1
中的所有元素逐个弹出并压入stack2
,然后再从stack2
中弹出一个元素。这个过程的时间复杂度是 O(n),其中 n 是队列中的元素数量。但是由于每个元素只会被移动一次,对于整个序列的操作,平均下来每个元素仍然是常数时间内完成的。 - 在均摊意义下,出队操作的时间复杂度可以视为 O(1)。
- 出队操作在最坏的情况下,需要将
-
查看队头元素 (
peek
):- 类似于出队操作,查看队头元素在最坏情况下也需要 O(n) 的时间,因为需要将元素从
stack1
移到stack2
。 - 在均摊意义下,查看队头元素的时间复杂度也可以视为 O(1)。
- 类似于出队操作,查看队头元素在最坏情况下也需要 O(n) 的时间,因为需要将元素从
-
判断队列是否为空 (
empty
):- 判断两个栈是否都为空是一个常数时间的操作,时间复杂度为 O(1)。