玩转数据结构（十一）两个队列实现栈

1、思路分析

• 现在有两个队列，queue1和queue2; 当要实现入栈操作时，将数据放入到不为空的队列中，要是两个队列都为空，默认先放入queue1队列中。如下图：a、b、c、d都放入到queue1队列中。

• 当要实现出栈操作时，将不为空的队列中的数据放入到另一个为空的队列中，直至剩下一个；而那一个数据就是要出队的数据。如下图：将queue1中的a、b、c依次转移到queue2中，只剩下d，而d就是要出队的数据，相当于栈顶数据。依次反复即可实现弹栈操作。

2、代码实现

import java.util.ArrayDeque;
import java.util.Queue;
/**
 * 两个队列实现栈
 * @param <E>
 */
public class QueueStack<E>{
    private Queue<E> queue1 = new ArrayDeque<E>();
    private Queue<E> queue2 = new ArrayDeque<E>();
    public boolean isEmpty() {
        return queue1.isEmpty() && queue2.isEmpty();
    }
    /**
     * 入队
     * @param e
     */
    public void enqueue(E e) {
        //若两个队列为空，则将数据优先放入queue1中
        if(isEmpty()) {
            queue1.add(e);
            return;
        }
        //若queue2为空，queue1不为空，将数据放入queue1
        if(queue2.isEmpty()) {
            queue1.add(e);
            return;
        }
        //若queue1为空，queue2不为空，将数据放入queue2
        if(queue1.isEmpty()) {
            queue2.add(e);
            return;
        }
    }
    /**
     * 出队
     * @return E
     */
    public E dequeue() {
        if(isEmpty()) {
            throw new RuntimeException("error! stack is empty!");
        }
        if(queue2.isEmpty()) {
            //将queue1中的数据放入到queue2，直至剩下一个数据，返回
            while(queue1.size() > 1) {
                queue2.add(queue1.poll());
            }
            return queue1.poll();
        }
        if(queue1.isEmpty()) {
            //将queue2中的数据放入到queue1，直至剩下一个数据，返回
            while(queue2.size() > 1) {
                queue1.add(queue2.poll());
            }
            return queue2.poll();
        }
        return null;
    }
}

3、测试代码

 public static void main(String[] args) {
        QueueStack<Integer> stackQueue = new QueueStack<Integer>();
        for (int i = 0; i < 10; i++) {
            System.out.println("入栈：" + i);
            stackQueue.enqueue(i);
            if(i % 3 == 2) {
                System.out.println("出栈：" + stackQueue.dequeue());
            }
        }
        while(!stackQueue.isEmpty()) {
            System.out.println("出栈：" + stackQueue.dequeue());
        }
    }

4、结果

入栈：0
入栈：1
入栈：2
出栈：2
入栈：3
入栈：4
入栈：5
出栈：5
入栈：6
入栈：7
入栈：8
出栈：8
入栈：9
出栈：9
出栈：7
出栈：6
出栈：4
出栈：3
出栈：1
出栈：0