Leetcode刷题笔记——剑指 Offer 09. 用两个栈实现队列（简单）

---

题目描述

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

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

方法一

复杂度分析

时间复杂度：对于插入和删除操作，时间复杂度均为 O(1)。插入不多说，对于删除操作，虽然看起来是 O(n) 的时间复杂度，但是仔细考虑下每个元素只会「至多被插入和弹出 stack2 一次」，因此均摊下来每个元素被删除的时间复杂度仍为 O(1)。
空间复杂度：O(n)。需要使用两个栈存储已有的元素。

C++代码

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() {
        // 如果stack2为空，则把stack1元素压入stack2完成倒序操作
        if(stack2.empty()){
            while(!stack1.empty()){
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        // 如果完成以上操作后stack2还为空，说明已经没有元素了，返回-1
        if(stack2.empty()){
            return -1;
        }else{ // 如果此时stack2中有元素则直接返回
            int deleteIteam = stack2.top();
            stack2.pop();
            return deleteIteam;
        }
    }
};

总结

注意C++中的pop返回为void，所以不能用push(pop())的写法。

参考链接

https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/solution/mian-shi-ti-09-yong-liang-ge-zhan-shi-xian-dui-l-3/