用栈来实现队列和用队列来实现栈(python)

原题链接:
225: Implement Stack using Queses
232: Implement Queue using Stacks

解题思路:就是用两个栈相互倒来实现队列的功能。也可以只用一个栈实现,但是个人感觉那样不如两个栈好理解,方便记忆。入队的时间复杂度是O(1),出队是O(n)
用栈来实现队列:

class MyQueue:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.stackA = []
        self.stackB = []

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

    def pop(self) -> int:
        """
        Removes the element from in front of queue and returns that element.
        """
        while len(self.stackA) > 1:
            self.stackB.append(self.stackA.pop())
        a = self.stackA.pop()
        while self.stackB:
            self.stackA.append(self.stackB.pop())
        return a 
    
    def peek(self) -> int:
        """
        Get the front element.
        """
        while self.stackA:
            self.stackB.append(self.stackA.pop())
        a = self.stackB.pop()
        self.stackA.append(a)
        while self.stackB:
            self.stackA.append(self.stackB.pop())
        return a 
            


    def empty(self) -> bool:
        """
        Returns whether the queue is empty.
        """
        return self.stackA == [] and self.stackB == []

用队列来实现栈:
需要注意的是在top这个操作里,要先把队列末尾的元素a拿出去,然后再放回队列中。

class MyStack:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.queueA = []
        self.queueB = []
        
    def push(self, x: int) -> None:
        """
        Push element x onto stack.
        """
        self.queueA.append(x)

    def pop(self) -> int:
        """
        Removes the element on top of the stack and returns that element.
        """
        if not self.queueA:
            return None
        while len(self.queueA) != 1:
            self.queueB.append(self.queueA.pop(0))
        self.queueA, self.queueB = self.queueB, self.queueA
        return self.queueB.pop()
        

    def top(self) -> int:
        """
        Get the top element.
        """
        while len(self.queueA) > 1:
            self.queueB.append(self.queueA.pop(0))
        a = self.queueA.pop(0)
        self.queueB.append(a)
        self.queueA, self.queueB = self.queueB, self.queueA
        return a

    def empty(self) -> bool:
        """
        Returns whether the stack is empty.
        """
        return self.queueA == []

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值