面试题09. 用两个栈实现队列

193 阅读1分钟

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

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/yo… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

  1. 一个栈用于入队,一个栈用于出队;
  2. 当出队栈为空时,将入队栈内容全部拷贝到出队栈;

使用Stack的方式来做这道题,会造成速度较慢;原因是因为Stack继承了Vector接口,而Vector底层是AbstractList,是一个数组,需要考虑空间扩容的问题了。可以使用LinkedList来做Stack的容器,因为LinkedList实现了Deque接口,所以Stack能做的事LinkedList都能做,其本身结构是个链表,扩容消耗少。

class CQueue {
    LinkedList<Integer> stack1;
    LinkedList<Integer> stack2;

    public CQueue() {
        stack1 = new LinkedList<>();
        stack2 = new LinkedList<>();
    }
    
    public void appendTail(int value) {
        stack1.add(value);

    }
    
    public int deleteHead() {
        if(stack2.isEmpty()){
            if(stack1.isEmpty()) { return -1;}
            while(!stack1.isEmpty()) {
                stack2.add(stack1.remove());
            }
            return stack2.remove();
        } else {
            return stack2.remove();
        }
    }
}