用两个栈来实现队列

题目描述

思路

栈的数据处理顺序是先进后出，队列的处理顺序是先进先出，想要用栈来实现队列，需要把栈本身的数据取出顺序反过来，需要两个栈来实现，栈1用来存入数据，然后把数据再弹出，弹出的同时再依次存入栈2，这样在栈2弹出数据的时候就能实现先入先出的顺序了。
可以用一串数字12345来模拟：

实现

用Python来实现的话，可以用列表来存储数据，作为栈，恰好列表也提供了pop方法来模拟栈的弹出行为。
需要注意的是，不需要再额外创造一个列表来存储栈2弹出的数据，直接将栈2弹出的数据依次保存到栈1中，那栈1最后弹出的数据顺序即为最后的先进先出的顺序了，这样充分利用栈1这个工具人的身份，节省了内存空间。

class CQueue:
    '''
    栈：先进后出
    队列：先进先出
    两个栈实现一个队列的思路：同一组数据两次入栈后再出栈
    需要栈1入栈，然后出栈，栈1出栈的数据再一次入栈到栈2中，把栈2的数据出栈再保存回栈1中，就能实现队列的先进先出
    '''

    def __init__(self):
        self.stack1 = []
        self.stack2 = []

    def appendTail(self, value: int) -> None:
        # 将栈1元素依次入栈2
        while self.stack1:
            self.stack2.append(self.stack1.pop())
        # 在尾部插入数据
        self.stack1.append(value)
        # 将栈2弹出的数据依次入栈1，把栈1当做拼接好后的队列
        while self.stack2:
            self.stack1.append(self.stack2.pop())
        return self.stack1

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

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