用两个栈实现队列
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
示例 1:
输入:
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出:[null,null,3,-1]
来源:力扣(LeetCode)
解题思路
两个栈A,B 模拟实现一个队列,A栈作为输入,B栈作为输出用。
1、入栈时直接加入到A栈中。
2、出栈时:
(1)B栈有数据,直接在B栈中pop
(2)B栈中没有数据,先将A栈的所有数据倒入B栈中,再从B栈中pop
(3)B栈和A栈中都没有数据,直接返回-1
class CQueue(object): def __init__(self): self._list1=list() self._list2=list() def appendTail(self, value): """ :type value: int :rtype: None """ self._list1.append(value) def deleteHead(self): """ :rtype: int """ _len1=len(self._list1) _len2=len(self._list2) if _len1==0 and _len2==0: return -1; elif _len2==0: for i in range(_len1): self._list2.append(self._list1.pop()) return self._list2.pop()