力扣:225. 用队列实现栈

题目:

使用队列实现栈的下列操作:

  • push(x) -- 元素 x 入栈
  • pop() -- 移除栈顶元素
  • top() -- 获取栈顶元素
  • empty() -- 返回栈是否为空

注意:

  • 你只能使用队列的基本操作-- 也就是 push to back, peek/pop from front, size, 和 is empty 这些操作是合法的。
  • 你所使用的语言也许不支持队列。 你可以使用 list 或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
  • 你可以假设所有操作都是有效的(例如, 对一个空的栈不会调用 pop 或者 top 操作)。

代码实现及详细注释:

class MyStack:

    def __init__(self):
        # 初始化栈,使用列表作为队列
        self.queue = []

    def push(self, x: int) -> None:
        # 入栈操作,直接在队尾添加元素
        self.queue.append(x)

    def pop(self) -> int:
        # 出栈操作,移除并返回栈顶元素
        if self.empty():  # 如果栈为空,返回 None
            return None
        for i in range(len(self.queue) - 1):
            self.queue.append(self.queue.pop(0))  # 将队首元素移动到队尾,使得栈顶元素位于队尾
        return self.queue.pop(0)  # 移除并返回队首元素,即为栈顶元素

    def top(self) -> int:
        # 获取栈顶元素,不移除元素
        ans = self.pop()  # 调用 pop 方法获取栈顶元素
        self.queue.append(ans)  # 将栈顶元素重新放入队列,保持栈不变
        return ans

    def empty(self) -> bool:
        # 判断栈是否为空
        return not self.queue  # 如果队列为空,返回 True


# Your MyStack object will be instantiated and called as such:
# obj = MyStack()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.top()
# param_4 = obj.empty()

复杂度分析·:

  • 时间复杂度: pop为O(n),其他为O(1)
  • 空间复杂度: O(n)

你可能感兴趣的:(python,算法,leetcode,数据结构,python,算法)