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

题目

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

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

示例 2: 输入:[“CQueue”,”deleteHead”,”appendTail”,”appendTail”,”deleteHead”,”deleteHead”] [[],[],[5],[2],[],[]] 输出:[null,-1,null,null,5,2]

提示:
1 <= values <= 10000
最多会对appendTail、deleteHead 进行10000次调用

题解

class CQueue {
    /**
     * 读栈
     */
    int[] read;
    /**
     * 写栈
     */
    int[] write;
    /**
     * 读索引
     */
    int readIndex;
    /**
     * 写索引
     */
    int writeIndex;

    public CQueue() {
        // 由题意最多10000个元素
        this.read = new int[10001];
        this.write = new int[10001];
        this.readIndex = -1;
        this.writeIndex = -1;
    }

    public void appendTail(int value) {
        this.write[++this.writeIndex] = value;
    }

    public int deleteHead() {
        if (this.writeIndex == -1 && this.readIndex == -1) {
            return -1;
        }

        if (this.readIndex == -1) {
            // 此时 读栈已经是一个队列
            while (this.writeIndex > -1) {
                this.read[++this.readIndex] = this.write[this.writeIndex--];
            }
        }

        return this.read[this.readIndex--];
    }
}