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

剑指 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 次调用

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

前置知识：

标签：

思路

1.自己的思路

两个栈分别维护进入的元素、弹出的元素，命名为inStack、outStack;

inStack的栈底模拟队首，栈顶模拟栈顶。每次添加元素到队尾----->添加元素到栈顶；

outStack用来维护弹出的队尾元素。每次弹出元素前都检查一下outStack是否为空，

• 如果为空就将inStack栈的元素依次弹入到OutStack中，此时outStack的栈顶元素即为队首元素（利用了栈可以翻转元素顺序的特性），然后弹出栈顶元素（如果此时outStack为空，返回-1）；
• 如果不为空，弹出outStack的栈顶元素即可。

代码

class CQueue {

    private Deque<Integer> inStack;  
    private Deque<Integer> outStack;

    public CQueue() {
        this.inStack = new LinkedList<>();
        this.outStack = new LinkedList<>();
    }

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

    public int deleteHead() {

        if(outStack.isEmpty()){
            if (!inStack.isEmpty()){
                while (!inStack.isEmpty()){
                    outStack.push(inStack.pop());
                }

            }
        }
        return outStack.isEmpty() ? -1 : outStack.pop();
    }
}

复杂度分析

• 时间复杂度：添加元素的时间复杂度为O(1), 弹出队首元素的时间复杂度为O(n), n为队列长度
• 空间复杂度：O(n)

总结

这道题目有一个需要注意的地方是如果你使用Stack的方式来做这道题，会造成速度较慢； 原因的话是Stack继承了Vector接口，而Vector底层是一个Object[]数组，那么就要考虑空间扩容和移位的问题了。 可以使用LinkedList来做Stack的容器，因为LinkedList实现了Deque接口，所以Stack能做的事LinkedList都能做，其本身结构是个双向链表，扩容消耗少。 但是我的意思不是像100%代码那样直接使用一个LinkedList当做队列，那确实是快，但是不符题意。 这样的优化之后，效率提高了40%，超过97%。