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

一、题目

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

示例一：

输入：

["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]

输出：

[null,null,3,-1]

示例二：

输入：

["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]

输出：

[null,-1,null,null,5,2]

二、题解

我刚开始没看懂题目的输入输出，看了官方的题解才明白，在此记录一下。
题目输入第一行的一个一个操作序列，第二行是对应的操作的参数。“CQueue”和"deleteHead"分别为初始化操作和删除队列第一个元素操作，不需要参数，所以第二行对应的位置为[]。“appendTail”为在队列尾部添加一个元素，需要一个参数，所以第二行对应位置为 [ * ] 。

思路：题目说利用两个栈实现队列操作，但是咱们只要实现功能即可。

方法一：利用两个Stack栈a和b模拟队列操作，入队的进栈a，出队的时候让栈a的元素全都放到栈b中，然后取栈b的栈顶元素即为模拟删除队列队头的首元素。

class CQueue {

    Stack<Integer> a,b;
    public CQueue() {
        a = new Stack<>();
        b = new Stack<>();
    }

    public void appendTail(int value) {
        a.push(value);
    }

    public int deleteHead() {

        if(!b.empty()){
            return b.pop();
        }

        if(a.empty()){
            return -1;
        }

        while (!a.empty()){
            b.push(a.pop());
        }
        return b.pop();
    }
}

方法二：利用双端队列Deque的特性当栈用，比较方便，但是需要记住对应的方法名

class CQueue {

    Deque<Integer> a;
    Deque<Integer> b;

    public CQueue() {
        a= new ArrayDeque<>();
        b= new ArrayDeque<>();
    }
    
    public void appendTail(int value) {
        a.push(value);
    }
    
    public int deleteHead() {
        if (b.isEmpty()) 
        		while (!a.isEmpty()) 
        			b.push(a.pop());
        if (b.isEmpty()) return -1;
        return b.pop(); 
    }
}