剑指offer 面试题9：用两个栈实现队列（java代码）

题目

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

class CQueue {

    public CQueue() {}
   
    public void appendTail(int value) {}
    
    public int deleteHead() {}
}

思路

• 栈是先进后出，队列是先进先出，用两个栈可以实现队列
• 先将数据压到一个栈中，然后再将这个栈中的数据压如第二个栈，那么第二个栈进行pop时就相当于队列的先进先出。
• 删除数据时直接对第二个栈进行出栈即可，插入数据时将数据插入第一个栈就相当于在队列尾部插入数据。

class CQueue {

    private Stack<Integer> stack1;
    private Stack<Integer> stack2;
    
    public CQueue() {
        stack1 = new Stack<>();
        stack2 = new Stack<>();
    }
    
    // 在队尾插入元素，直接在stack1中插入即可。
    public void appendTail(int value) {
        stack1.push(value);
    }
    
    public int deleteHead() {
        //当stack2的容量为空时，如果stack1里面不为空，则将stack1中的数据压如stack2中
        if (stack2.isEmpty()) {
            if (stack1.isEmpty()) return -1;
            
            while (stack1.size() > 0) {
                stack2.push(stack1.pop());
            }
        }
        //如果stack2的容量不为空时，要想对队列进行删除头部节点操作，直接对stack2进行出栈即可。
        return stack2.pop();
    }
}

复杂度分析

• 时间复杂度：对于插入和删除操作，时间复杂度均为 O(1)。对于删除操作，每个元素至多被插入和弹出 stack2 一次，因此每个元素被删除的时间复杂度仍为 O(1)
• 空间复杂度：O(n)。需要使用两个栈存储已有的元素