欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

面试题09. 用两个栈实现队列【LeetCode剑指offer】

程序员文章站 2022-06-17 17:47:04
...

题目:

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

面试题09. 用两个栈实现队列【LeetCode剑指offer】

思路

  1. 两个栈,一个size(队列中数据的数量)
  2. 入队列直接压入栈1,并且size++;
  3. 出队列时,先判断size是否为0,若size 为 0,则说明队列空,返回-1;
  4. 否则直接取出栈2的栈顶元素,若栈2为空,则把栈1 的所有元素一次性出栈压入栈2,此时栈2,按顺序弹出则是队列出队列的顺序,先进先出;出队列时,要相应的的size–;

实现:

class CQueue {
    Stack<Integer> stack1;
    Stack<Integer> stack2;
    int size;


    public CQueue() {
        stack1 = new Stack();
        stack2 = new Stack();
        size = 0;
    }

    public void appendTail(int value) {
        stack1.push(value);
        size++;
    }

    public int deleteHead() {
        if (size == 0) {
            return -1;
        }
        if (stack2.empty()) {
            while (!stack1.empty()) {
                stack2.push(stack1.pop());
            }
        }
        size--;
        return stack2.pop();
    }
}