问题描述:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
示例
输入:
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出:[null,null,3,-1]
输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]
解析
刚开始没看懂题目。。。
输入: ["CQueue","appendTail","deleteHead","deleteHead"]
这里是要执行的方法,从左到右执行。[[],[3],[],[]]
对应上面的方法,是上面方法的参数。CQueue和deleteHead方法不需要指定数字,只有添加才需要指定数字
1.创建队列,返回值为null
2.将3压入栈,返回值为null
3.将栈底的元素删除,也就是消息队列中先进来的元素,所以是deleteHead,返回该元素的数值,所以为3
4.继续删除栈底的元素,但是没有元素了,所以返回-1
所以就有了下面的输出 输出:[null,null,3,-1]
解决方案:
明确一下栈和队列含义:
栈:在栈顶插入、删除【LIFO】后进先出
队列:对尾插入,对首删除【FIFO】先进先出
- 栈无法实现队列功能: 栈底元素(对应队首元素)无法直接删除,需要将上方所有元素出栈。
- 双栈可实现列表倒序: 设有含三个元素的栈 A = [1,2,3] 和空栈 B = []。若循环执行 A 元素出栈并添加入栈 B ,直到栈 A 为空,则 A = [], B = [3,2,1],即 栈 B 元素实现栈 A 元素倒序 。
- 利用栈 B删除队首元素: 倒序后,B执行出栈则相当于删除了 A的栈底元素,即对应队首元素。
加入队尾 appendTail()函数: 将数字 val 加入栈 A 即可。
删除队首deleteHead()函数: 有以下三种情况。
- 当栈 B 不为空: B中仍有已完成倒序的元素,因此直接返回 B 的栈顶元素。
- 当 A 为空: 即两个栈都为空,无元素,因此返回 -1 。
- 否则: 将栈 A 元素全部转移至栈 B 中,实现元素倒序,并返回栈 B 的栈顶元素。
code
class CQueue:
def __init__(self):
self.stack_A,self.stack_B=[],[]
def appendTail(self, value: int) -> None:
self.stack_A.append(value)
def deleteHead(self) -> int:
#这句可能会落下:考虑到b里有倒序完成但还没pop()的元素时候,应该直接返回b.pop(),等b为空a不空的时候,才进行先倒叙后b.pop()的操作。
if(self.stack_B):return self.stack_B.pop()
if (not self.stack_A):return -1
while(self.stack_A):
self.stack_B.append(self.stack_A.pop())
return self.stack_B.pop()
results
------------2020/12/31重新了解了一下栈和队列!------------