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

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

1、 问题描述

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

https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/

2、 实例分析

两个栈stack1 和stack2实现队列，这道题的意图是要求我们操作这两个“先进后出”的栈实现一个“先进先出”的队列CQueue。
我们通过一个具体的例子来分析往该队列插入和删除元素的过程。
首先插入–个元素a，不妨先把它插入到 stack1，此时stack1中的元素有{a}，stack2为空。 再压入两个元素b和c，还是插入到stack1中，此时stack1中的元素有{a, b,c}，其中c位于栈顶，而stack2仍然是空的（如图(a)所示)。 这个时候我们试着从队列中删除一个元素。按照队列先入先出的规则，由于a比 b、c先插入到队列中，最先被删除的元素应该是a。元素a存储在stack1中，但并不在栈顶上，因此不能直接进行删除。注意到stack2我们还一直没有使用过，现在是让stack2发挥作用的时候了。如果我们把stack1中的元素逐个弹出并压入stack2，元素在stack2中的顺序正好和原来在stack1中的顺序相反。因此经过3次弹出stack1和压入 stack2操作之后,stack1为空，而stack2中的元素是{c,b,a}，这个时候就可以弹出stack2的梭顶a了。此时的stackl为空，而stack2的元素为{c,b}，其中b在栈顶(如图 (b）所示)。 如果我们还想继续删除队列的头部应该怎么办呢?剩下的两个元素是b和c， b比c早进入队列，因此b应该先删除。而此时b恰好又在栈顶上，因此直接弹出stack2的栈顶即可。这次弹出操作之后，stack1 中仍然为空，而stack2为{c}（如图 (c）所示)。
从上面的分析中我们可以总结出删除一个元素的步骤;当stack2中不为空时，在stack2中的栈顶元素是最先进入队列的元素，可以弹出。如果stack2为空时，我们把stack1中的元素逐个弹出并压入stack2。由于先进入队列的元素被压到stackl 的底端，经过弹出和压入之后就处于stack2的顶端了，又可以直接弹出。
接下来再插入一个元素d。我们还是把它压入stack1（如图（d）所示),这样会不会有问题呢?我们考虑下一次删除队列的头部stack2不为空，直接弹出它的栈顶元素c （如图 (e)所示)。而c的确是比d先进入队列，应该在d之前从队列中删除，因此不会出现任何矛盾。

3、 分析总结

定义两个stack，分别是stack1和stack2，队列的push和pop是在两侧的，push操作很简单，只需要在stack1上操作，而pop操作时，先将stack1的所有元素push到stack2中，然后stack2的pop返回的元素即为目标元素，只要stack2不为空下次删除队头元素时可以从Stack2中直接出栈，只有当栈2为空栈1不为空时才将栈1的元素出栈压入到栈2中。

4、python实现

class CQueue2:
    def __init__(self):
        self.stack1 = []
        self.stack2 = []
    def appendTail(self, value: int) -> None:
        self.stack1.append(value)
    def deleteHead(self) -> int:
        if len(self.stack2)<=0:  # 栈2为空循环--只有当栈2为空的时候才将栈1中的元素 压入栈2中
            while self.stack1:  # 栈1不为空循环
                self.stack2.append(self.stack1.pop())  # 栈1中的元素 压入栈2中  栈1的栈底变成栈2的栈顶
        if not self.stack2:  # 如果栈1栈2都为空 则返回-1
            return -1
        else:
            result = self.stack2.pop()  # 当栈2为不空的时候，栈2的栈顶元素直接出栈
        return result

# Your CQueue object will be instantiated and called as such:
obj = CQueue2()
obj.appendTail(1)
obj.appendTail(2)
obj.appendTail(3)
obj.appendTail(4)
param_2 = obj.deleteHead()
print(param_2)
param_2 = obj.deleteHead()
print(param_2)
param_2 = obj.deleteHead()
print(param_2)
param_2 = obj.deleteHead()
print(param_2)
param_2 = obj.deleteHead()
print(param_2)
obj.appendTail(1)
obj.appendTail(2)
obj.appendTail(3)
obj.appendTail(4)
param_2 = obj.deleteHead()
print(param_2)
param_2 = obj.deleteHead()
print(param_2)