剑指 Offer 09. 用两个栈实现队列(力扣)

一：题目

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

示例 1：

输入：

["CQueue","appendTail","deleteHead","deleteHead","deleteHead"]

[[],[3],[],[],[]]

输出：[null,null,3,-1,-1]

示例 2：

输入：

["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]

[[],[],[5],[2],[],[]]

输出：[null,-1,null,null,5,2]

二：解题思路

第一步：

用两个栈实现尾部插入和头部删除，首先我们就需要在全局定义两个栈，分别用作插入栈（inStack）和删除站（outStack）

stack<int> inStack, outStack;

第二步：实现插入

实现队尾的插入，直接运用push函数即可

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

第三步：实现删除

第一种情况：栈为空

这种情况只需要判断两个栈为空，返回-1即可

 if(outStack.empty())
        {
            if(inStack.empty())
            {
                return -1;
            }
        }

第二种情况：栈中有数字需要删除

假如，我们现在先后插入了5和2两个数字在我们的inStack里面

因为是从队头开始删除，所以我们是先删除5然后删除2

那么只需要将inStack里面的数据一个一个压入outStack的栈即可

总代码：

class CQueue {
    private:
     stack<int> inStack, outStack;//设置的两个栈，进栈和出栈
public:
    CQueue() {
    }
    
    void appendTail(int value) {//直接push压入
        inStack.push(value);
    }
    
    int deleteHead() {
        if(outStack.empty())
        {
            if(inStack.empty())//都为空返回-1
            {
                return -1;
            }
            else
            {
                while(!inStack.empty())//将inStack的数据全部转入outStack中
                {
                    outStack.push(inStack.top());
                    inStack.pop();
                }
            }
        }
        int value=outStack.top();//将outStack中的数据一个一个出去
        outStack.pop();
        return value;
    }
};