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

题目描述

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

解析

首先，对题目中举的例子要正确理解
看第一个栗子：
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]表示要执行的方法，
[[],[3],[],[]]从左到右依次表示上面的方法的参数,CQueue和deleteHead方法不需要指定数字，只有添加才需要指定数字

• "CQueue"表示新建一个队列，返回值为null
• "appendTail"表示把3压入栈，返回值为null
• "deleteHead"表示弹出一个元素，由于刚才压进去一个元素3，因此这里的返回值是3
• "deleteHead"弹出一个元素，由于此时队列为空，因此返回值为-1
• 因此输出是[null, null, 3, -1]

接着要对栈和队列的特性进行分析：
栈先进后出，而队列是先进先出，也就是说对于【5 3 8 2】，

• 入栈顺序是【5 3 8 2】，出栈顺序是【2 8 3 5】
• 入队顺序是【5 3 8 2】，出队顺序还是【5 3 8 2】

我们使序列【5 3 8 2】入栈，为了得到【5 3 8 2】的结果，
我们定义两个栈：append_stack(stack1)、delete_stack(stack2)
python中用列表来实现栈和队列

class CQueue:

    def __init__(self):
        self.append_stack = []
        self.delete_stack = []


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

#我们规定：必须当stack2中的元素pop完之后，也就是stack2为空时，再插入stack1中的元素；
#每次要出队首元素的时候，tack2有元素就pop  stack2中的元素，stack2没有元素了，就把stack1中的元素全部压入stack2中再弹出（顺序相反）；stack1、stack2都为空，说明没有元素了，则返回-1
    def deleteHead(self) -> int:
   		 # （1）当stack2不为空时，弹出stack2中的元素
        if self.delete_stack:
            return self.delete_stack.pop()
         # （2）当stack2为空、stack1不为空时，依次将stack1中的元素弹出，再压入stack2中
        while self.append_stack:
            self.delete_stack.append(self.append_stack.pop())
         # 当stack1、stack2均为空时，说明没有可弹出的元素，返回-1
        if not self.delete_stack:
            return -1
        return self.delete_stack.pop()
# Your CQueue object will be instantiated and called as such:
# obj = CQueue()
# obj.appendTail(value)
# param_2 = obj.deleteHead()