玩转数据结构(十一)两个队列实现栈

1、思路分析

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

玩转数据结构(十一)两个队列实现栈_第1张图片

  • 当要实现出栈操作时,将不为空的队列中的数据放入到另一个为空的队列中,直至剩下一个;而那一个数据就是要出队的数据。如下图:将queue1中的a、b、c依次转移到queue2中,只剩下d,而d就是要出队的数据,相当于栈顶数据。依次反复即可实现弹栈操作。
玩转数据结构(十一)两个队列实现栈_第2张图片

2、代码实现

import java.util.ArrayDeque;
import java.util.Queue;
/**
 * 两个队列实现栈
 * @param 
 */
public class QueueStack{
    private Queue queue1 = new ArrayDeque();
    private Queue queue2 = new ArrayDeque();
    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 stackQueue = new QueueStack();
        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

你可能感兴趣的:(数据结构与算法)