【LeetCode每日一题】2020.6.30 剑指 Offer 09. 用两个栈实现队列

剑指 Offer 09. 用两个栈实现队列

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 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]

分析：

​ 栈，先进后出。队列，先进先出。如果要用两个栈来模拟队列，即保证第一个栈栈底的元素，在需要时（队列popLeft操作）在第二个栈中符合先进先出的顺序。想象两个箱子装书，只要将第一个箱子的书一次拿出，并放进第二个箱子，正好就可以将书的顺序翻转。

​ 如果第二个栈stack2中有多个元素，下一次进行删除操作可以直接在stack2中pop。所以每次deleteHead之前进行一次判断即可。

代码（Python）：

class CQueue:

    def __init__(self):
        self.stack = []
        self.buffer = []

    def appendTail(self, value: int) -> None:
        self.stack.append(value)

    def deleteHead(self) -> int:
        if not self.buffer:
            while self.stack:
                self.buffer.append(self.stack.pop())
            if not self.buffer:
                return -1
        return self.buffer.pop()