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

题目内容

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 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后依次加入到第二栈里面，全部导入后，第二个栈的顺序就是删除的顺序。
  

时间复杂度： 每个元素至多被插入和删除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;
        }
    }
};