【小白用python刷Leetcode】剑指 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]

提示：

• 1 <= values <= 10000
• 最多会对 appendTail、deleteHead 进行 10000 次调用

解题思路

又刷到老面孔，又不禁笑出声来，这两天欧气这么爆的吗哈哈。面试必会的一道题，剑指offer里也很靠前，熟的我都不想打题目描述了，还要调格式好麻烦的说，下次可以考虑直接截图。

思路呢也就是常规思路，反正题目要求里都把思路限定死了：用两个栈实现队列。简单来说就是建两个栈，stack1负责元素入队列，stack2负责元素出队列，而且两个栈各自只负责一种职能，不能混。需要强调的一点是在出栈时，如果stack2为空，要用while循环把stack1内的所有元素都拿过来，不然再有元素入栈时，顺序会乱。细节的话也没啥，就是题目里有个小要求，如果队列里没有元素，调用出队列时要返回-1，别忘了加上就好。

题解代码：

class CQueue:

    def __init__(self):
        self.stack1 = []  # 入队列
        self.stack2 = []  # 出队列

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

    def deleteHead(self) -> int:
        if not self.stack1 and not self.stack2:
            return -1
        if not self.stack2:
            while self.stack1:
                self.stack2.append(self.stack1.pop())       
        return self.stack2.pop()



# Your CQueue object will be instantiated and called as such:
# obj = CQueue()
# obj.appendTail(value)
# param_2 = obj.deleteHead()

以上就是我，一个正儿八经的小白（大神们通过看代码应该也感觉出来了），对这道题的理解，欢迎诸位指正讨论，感谢阅读。

原题链接：

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