剑指offer | 09. 用两个栈实现队列

系统管理员 2022-12-21 14:55 72阅读 0赞

**题目内容**

> 用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 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 次调用

--------------------

**思路：**

*  维护两个栈，第一个栈支持插入操作，第二个栈支持删除操作。
 *  插入操作时都将元素插入到第一个栈中；
 *  当要进行删除操作时，若第二个栈为空，将第一个栈的元素pop后依次加入到第二栈里面，全部导入后，第二个栈的顺序就是删除的顺序。  
    ![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQzMTQ1OTI2_size_16_color_FFFFFF_t_70_pic_center]

**时间复杂度：** 每个元素至多被插入和删除1次，所以时间复杂度为O（1）.  
**空间复杂度：** 需要两个栈存储已有的元素，所以是O（n）。

**完整代码**：

class CQueue { 
    stack<int> stack1,stack2;
    public:
        CQueue() { 
            while(!stack1.empty()){ 
                stack1.pop();
            }
    
            while(!stack2.empty()){ 
                stack2.pop();
            }
        }
        
        void appendTail(int value) { 
            stack1.push(value);
        }
        
        int deleteHead() { 
            //如果第二个栈为空
            if(stack2.empty()){ 
                while(!stack1.empty()){ 
                    stack2.push(stack1.top());
                    stack1.pop();
                }
            }
            if(stack2.empty()){ 
                return -1;
            }else{ 
                int deleteItem=stack2.top();
                stack2.pop();
                return deleteItem;
            }
        }
    };

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQzMTQ1OTI2_size_16_color_FFFFFF_t_70_pic_center]: /images/20221120/9f04c38c0e8b49afa0472075338f362f.png