队列的应用之队列实现栈

借助两个栈来实现

先进后出 后进先出--------> 先进先出

一个栈放   另一个栈临时存放

package p2.线性结构;

import p1.接口.Queue;

import java.util.Iterator;

//栈实现队列
public class StackToQueue {
    public static void main(String[] args) {
        QueueImplByStack queue = new QueueImplByStack<>();
        for (int i = 0; i <= 5; i++) {
            queue.offer(i);
        }
        System.out.println(queue);
        System.out.println(queue.poll());
        System.out.println(queue);
    }
}

class QueueImplByStack implements Queue {
    private ArrayStack stackA;
    private ArrayStack stackB;

    public QueueImplByStack() {
        stackA = new ArrayStack<>();
        stackB = new ArrayStack<>();
    }

    @Override
    public void offer(E element) {
        stackA.push(element);
    }

    @Override
    public E poll() {
        if (isEmpty()) {//是空
            throw new IllegalArgumentException("queue is null");
        }
        while (stackA.size() != 1) {//只要stackA有一个，则弹出来放stackB
            stackB.push(stackA.pop());
        }
        E ret = stackA.pop();//出列
        while (!stackB.isEmpty()) {//只要stackB不为空，则弹放stackA
            stackA.push(stackB.pop());
        }
        return ret;
    }

    @Override
    public E peek() {
        if (isEmpty()) {
            throw new IllegalArgumentException("queue is null");
        }
        while (stackA.size() != 1) {
            stackB.push(stackA.pop());
        }
        E ret = stackA.peek();
        while (!stackB.isEmpty()) {
            stackA.push(stackB.pop());
        }
        return ret;
    }

    @Override
    public boolean isEmpty() {
        return stackA.isEmpty();
    }

    @Override
    public void clear() {
        stackA.clear();
    }

    @Override
    public int size() {
        return stackA.size();
    }

    @Override
    public Iterator iterator() {
        return stackA.iterator();
    }

    @Override
    public String toString() {
        return stackA.toString();
    }
}