剑指offer学习笔记 | 用两个栈实现队列

题目

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

示例

 输入： ["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]

[[],[],[5],[2],[],[]]

1.创建队列，返回值为null

2.删除栈底的元素，但是没有元素，所以返回-1

3.把5压入栈，返回null

4.把2压入栈，返回null

5.删除栈底的一个元素，也就是消息队列中先进来的元素，所以是deleteHead，就是最先进来的5，返回值为5，

6.删除栈底的一个元素，就是后进来的2，返回值为2

因此，输出结果为

输出：[null,-1,null,null,5,2]

解决思路

使用两个栈，一个栈负责进入，另一个栈负责出。当有元素加入时，第一个栈进入。当有元素出时，分三种情况讨论：

1、两个栈都为空，此时队列为空；

2、第一个栈有元素，第二个栈没有元素，把第一栈的元素全部出栈放入第二个栈，第二个栈出栈；

3、第二个栈还有元素，第二个栈出栈。

代码实现

class CQueue {
    Stack <Integer> in;
    Stack <Integer> out;
    public CQueue() {
        in = new Stack <Integer>();
        out = new Stack <Integer>();
    }
    
    public void appendTail(int value) {
        in.push(value);
    }
    
    public int deleteHead() {
        if(in.empty() && out.empty()) return -1;
        if(out.empty()){
            while(!in.empty()){
               out.push(in.pop());
            }
        }
        return out.pop();
    }
}

/**
 * Your CQueue object will be instantiated and called as such:
 * CQueue obj = new CQueue();
 * obj.appendTail(value);
 * int param_2 = obj.deleteHead();
 */