剑指offer leetcode python版 面试题09 用两个栈实现队列

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

我们在学习数据结构的时候知道，栈是先进后出，队列是先进先出。要用两个栈实现队列的添加数据和删除数据的操作。定义两个栈，栈a和栈b，栈a用来加入数据，实现队列的添加数据功能，另一个用来输出数据，实现队列的输出功能。但是要注意一点，当栈b为空的时候，必须先把栈a的所有数据全部出栈到栈b中，否则就会造成数据顺序混乱。我们可以用1，2，3，4四个数字在纸上演算一下就能轻松得到结果。

python中的列表可以模拟栈的功能，append和pop方法的使用效果如下：

a = [1,2,3,4]
a.append(5)
print(a)
a.pop()
print(a)
输出：[1, 2, 3, 4, 5]
输出:[1, 2, 3, 4]

接下来看本题目的代码：

class CQueue:

    def __init__(self):
        self.A,self.B = [],[]

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

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




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

上文说的a、b功能是为了说明解法，实际中怎么使用都看个人习惯。
首先定义了两个列表用来模拟栈的功能。在队列的尾部添加数据的话就直接用一个栈进行添加。

在队列的头部出队的话，如果用于出队的栈里面有数据，那直接进行出栈操作就可以了，若其里面为空，就要把另一个栈里面的数据全部出栈到本栈，这是为了保证数据的正确输出顺序。

有一说一，两个栈实现一个队列有点负负得正的感觉。

接下来看一下代码的判断逻辑：
如果A不空，那么直接出栈一次就行；
如果B也为空，那么返回-1，因为整个结构里面已经没有数据了；
再接下来，程序走到这说明A空而B不空，就需要把A里面的数据全部搞到B里面去。