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

题目描述

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

实现

队列特性：先进先出
栈特性：先进后出

用两个栈实现队列的先进先出，其中一个栈 stackOut 用来出，另一个栈 stackIn 用来进。
当出栈的时候，如果 stackOut 为空，那么需要将 stackIn 中所有元素逐个 pop 并同时压入 StackOut

// c++
class CQueue
{
public:
    CQueue()
    {
        while (!stackIn.empty())
            stackIn.pop();

        while (!stackOut.empty())
            stackOut.pop();
    }

    void appendTail(int value)
    {
        stackIn.push(value);
    }

    int deleteHead()
    {
        if (stackOut.empty())
        {
            if (stackIn.empty())
                return -1;
            
            while(!stackIn.empty())
            {
                stackOut.push(stackIn.top());
                stackIn.pop();
            }
        }

        
        int deleteValue = stackOut.top();
        stackOut.pop();
        return deleteValue;
        
    }
private:
    stack<int> stackIn, stackOut;
};

# python
class CQueue:
    def __init__(self):
        self.stack_in = []
        self.stack_out = []

    def append_tail(self, value: int) -> None:
        self.stack_in.append(value)

    def delete_head(self) -> int:
        if not self.stack_out:
            if not self.stack_in:
                return -1
            else:
                while self.stack_in:
                    self.stack_out.append(self.stack_in.pop())
        return self.stack_out.pop()