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

题目描述

思路分析

维护两个栈，第一个栈支持插入操作，第二个栈支持删除操作。

代码展示

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;
        }
    }
};

结果分析

时间复杂度为 O(1)
空间复杂度:O(n)。需要使用两个栈存储已有的元素。