在力扣上看到一道题:
题目:用两个栈实现队列
描述:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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()
注:这只是一种最好理解的写法,实际可以优化。