Leetcode-剑指offer-01-用两个栈实现队列

用两个栈实现队列

leetcode链接：https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/

题目如下：

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

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

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

解题思路则是用两个栈来模拟队列，其中栈stk1用来保存队列里的数据，stk2用来负责在删除的时候作为数据的临时储备用的栈

有两种解题思路，简单直接的一种是算法A， 加入一些优化后为算法B

算法A思路如下

1. 添加数据的时候直接添加到stk1的最顶端即可
2. 删除数据的时候：
   1. 如果stk1为空，说明队列为空，返回-1
   2. 将stk1弹出到stk2中，并取得stk2的顶部的数字，此为queue的Head，保存其当作要删除的值，同时pop掉它，然后再将stk2的弹出到stk1中，这样就可以保证删除的是栈stk1原有的底部的元素（即队列的Head）

算法的Code如下;

class CQueue {
public:
    stack<int> stk1;
    stack<int> stk2;

    CQueue() {

    }
    
    void appendTail(int value) {
        stk1.push(value);
    }
    
    int deleteHead() {
        if (stk1.empty()) return -1;
        while(stk1.size()>1)
        {
            stk2.push(stk1.top());
            stk1.pop();
        }
        int res = stk1.top();
        stk1.pop();
        while(stk2.size())
        {
            stk1.push(stk2.top());
            stk2.pop();
        }
        return res;
    }
};

算法B则是在算法A基础上面的一个优化，它的优化思路如下：

• 每次添加元素同样只添加到stk1中，当需要删除的时候，查看skt2是否为空，如果为空，则弹出stk1的元素到stk2中
• 可以观察到，上述操作保证queue前面的元素总是在stk2中，后面的元素总是在stk1中
• 每次stk1弹出元素到stk2后，stk2保存元素的顺序与stk1的相反，则stk2的顶部的元素保存的就是queue前面的元素，则每次deleteHead直接delete stk2的top即可

算法code 如下：

class CQueue {
public:
    stack<int> stk1;
    stack<int> stk2;

    CQueue() {

    }
    
    void appendTail(int value) {
        stk1.push(value);
    }
    
    int deleteHead() {
        if (stk1.empty()&&stk2.empty()) return -1;
        if (stk2.empty())
        {
            while(!stk1.empty())
            {
                stk2.push(stk1.top());
                stk1.pop();
            }
        }
        int res = stk2.top();
        stk2.pop();
        return res;
    }
};

最后我的源码都整理在了github上，剑指offer专题会分类整理，github链接如下，如果对您有帮助，请帮忙在github中点击收藏

github链接： https://github.com/wjlpku/Leetcode-interview