[Leetcode Daily] 232. 用栈实现队列

Description

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列的支持的所有操作(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(双端队列)来模拟一个栈,只要是标准的栈操作即可。

思路

直接一把梭。这里使用collections.deque
简要思路:

  1. push:直接stack1.push
  2. pop:如果stack2有元素,直接stack2.pop;没有,把stack1的元素一个个pop到stack2中再pop。
  3. enpty:not(stack1 or stack2)
  4. peek: 如果stack2有元素,返回stack2[0],没有,返回stack1[-1]
    剩下就是一些实现问题需要注意。

代码

class MyQueue:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.stack1=deque()
        self.stack2=deque()


    def push(self, x: int) -> None:
        """
        Push element x to the back of queue.
        """
        self.stack1.appendleft(x)



    def pop(self) -> int:
        """
        Removes the element from in front of queue and returns that element.
        """
        if self.stack2:
            return self.stack2.popleft()
        while self.stack1:
            self.stack2.appendleft(self.stack1.popleft())
        return self.stack2.popleft()


    def peek(self) -> int:
        """
        Get the front element.
        """
        if self.stack2:
            return self.stack2[0]
        elif self.stack1:
            return self.stack1[-1]


    def empty(self) -> bool:
        """
        Returns whether the queue is empty.
        """
        return not(self.stack1 or self.stack2)

执行用时:40 ms, 在所有 Python3 提交中击败了59.21%的用户
内存消耗:15.1 MB, 在所有 Python3 提交中击败了12.45%的用户
用了deque怎么还这么拉跨。。。
打卡。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值