刷题__6_用两个栈实现队列——泪崩,为什么我连题目都看不懂?

在力扣上看到一道题:
题目:用两个栈实现队列

描述:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
示例 1:

输入:
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出:[null,null,3,-1]

示例 2:

输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]

提示:

1 <= values <= 10000
最多会对 appendTail、deleteHead 进行 10000 次调用

一开始真的想了好久,我就寻思着:栈队列我都懂,为什么这个题目还是看不懂呢?
其实原理很简单,首先我们得知道题目的意思:删除操作中若有数据则返回删除的,没有数据则返回 -1,其他操作不返回数据,那其实就是null。
输入中第二个列表指的是对应的数据,只有添加的时候才有对应数字,删除和CQueue操作对应的数据为空。

["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]

表示如下:

  • 第一步:使用 CQueue 开始,不输出即 null
  • 第二步:调用一次appendTail,对应添加 3,不输出即 null
  • 第三步:调用一次deleteHead,此时删除 3,输出 3
  • 第四步:调用一次deleteHead,此时无数据可被删除,输出 -1
    所以输出为:[null,null,3,-1],(留下了没有技术的泪水)。

思考:使用两个栈来模拟队列(尾部进,头部出)元素的添加、删除操作,我们使用一个主栈来保存当前所有元素,当删除时再借用辅助栈。

解题:

class CQueue:

    def __init__(self):
        self.A, self.B = [], [] //两个栈 A为主 B为辅

    def appendTail(self, value: int) -> None:
        while self.A:
            self.B.append(self.A.pop()) // 添加时 先把A得转入B 再添加进A
        self.A.append(value)
        while self.B:
            self.A.append(self.B.pop()) // 暂存在B中的数据转回来
        return self.A

    def deleteHead(self) -> int:
        if not self.A: return -1 // 删除时A为空的话返回-1
        return self.A.pop() // A不为空的话出栈并输出


# Your CQueue object will be instantiated and called as such:
# obj = CQueue()
# obj.appendTail(value)
# param_2 = obj.deleteHead()

注:这只是一种最好理解的写法,实际可以优化。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值