用两个栈实现一个队列

Posted 2020-10-01

思路:

使用两个栈实现一个队列，栈A用作插入栈，栈B用作弹出栈。

队列的特点为先进先出。

1. 入列时，直接入栈A；
2. 出队列时先判断栈B是否为空，若为空，则依次弹出栈A中的数据入栈B，再弹出栈B数据；若不为空直接弹出栈B数据。

代码如下：

public class QueueByTwoStack<E> {

    private Stack<E> s1 = new Stack<>();
    private Stack<E> s2 = new Stack<>();
  
    public  void enqueue(E e) {
        s1.push(e);
    }

    public E dequeue() {
        if (s2.isEmpty()) {
            while (!s1.isEmpty()) {
                s2.push(s1.pop());
            }
        }
        return s2.pop();
    }

    public boolean isEmpty() {
        return s1.isEmpty() && s2.isEmpty();
    }

}