【力扣刷题】剑指 Offer 09. 用两个栈实现队列

1.题目：

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

示例 1：
输入：
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出：[null,null,3,-1]

示例 2：
输入：
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出：[null,-1,null,null,5,2]
提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.思路：

利用二个stack来实现队列，队列的特性是先进先出（fifo），而栈的特性是后进先出。用一个栈为主要栈stack1负责加入元素，待要出队列时，需找到队头弹出。这时需要辅助栈stack2负责找出队头弹出。将stack1的元素全部导入辅助栈stack2中此时stack2的栈顶即为队头，队头弹出以后重新装入stack1中回到了原来的状态。实现代码如下：

class CQueue {
public:
        stack<int> num1;
        stack<int> num2;
    CQueue() {

    }

    void appendTail(int value) {
         num1.push(value);
    }
    
    int deleteHead() {
        int temp;
        int temp1;
        while(!num1.empty()){
            temp=num1.top();
            num2.push(temp);
            num1.pop();
        }
        if(num2.empty()){
            return -1;
        }else{
            temp1 = num2.top();
            num2.pop();
            while(!num2.empty()){
                temp=num2.top();
                num1.push(temp);
                num2.pop();
            }
            return temp1;
        }    
    }

3.优化：

经过查阅资料参考他人代码，原来的代码实现起来比较死板不够灵活。优化的主要地方在于辅助栈负责弹出，主栈负责压入。相当如把一个队列分成了二个部分，当辅助栈为空的时候，需将主栈中尽可能多的元素全部放入辅助栈中。不为空时继续负责元素的弹出。这样的优化在一定程度上对一些非必要的步骤简化了，较为灵活。实现代码如下：

class CQueue {
public:
        stack<int> num1;
        stack<int> num2;
    CQueue() {
        while(!num1.empty()){
            num1.top();
        }
        while(!num2.empty()){
            num2.top();
        }
    }

    void appendTail(int value) {
         num1.push(value);
    }
    
    int deleteHead() {
        int temp;
        if(num1.empty()&&num2.empty()){
            return -1;
        }
        if(!num2.empty()){
            temp = num2.top();
             num2.pop();
             return temp;
        }else{
            while(!num1.empty()){
                temp = num1.top();
                num2.push(temp);
                num1.pop();
            }
            temp = num2.top();
            num2.pop();
            return temp;
        }
    }
};