用两个栈实现队列

要求：编写一个类，用两个栈实现队列，支持队列的基本操作(add、poll、peek)。

add 增加一个元索如果队列已满，则抛出一个IIIegaISlabEepeplian异常
remove 移除并返回队列头部的元素如果队列为空，则抛出一个NoSuchElementException异常
element 返回队列头部的元素如果队列为空，则抛出一个NoSuchElementException异常
offer 添加一个元素并返回true 如果队列已满，则返回false
poll 移除并返问队列头部的元素如果队列为空，则返回null
peek 返回队列头部的元素如果队列为空，则返回null
put 添加一个元素如果队列满，则阻塞
take 移除并返回队列头部的元素如果队列为空，则阻塞

思路：栈的特点：先进先出；队列的特点：先进后出；用两个栈正好顺序反过来，可以实现队列的操作；

方法：一个压入栈：只完成push操作，stackPush

一个弹出栈：只完成pop操作，stackPop

【重要】要完成如上操作，必须满足：

stackPush 执行pop 操作至stackPop时，stackPush必须全部倒出 & stackPop必须为null

public class Problem02_TwoStacksImplementQueue {
    public static class TwoStacksQueue {
        public Stack<Integer> stackPush;
        public Stack<Integer> stackPop;

        public TwoStacksQueue() {
            stackPush = new Stack<Integer>();
            stackPop = new Stack<Integer>();
        }

        /*
         * add操作： stackPush中执行push操作
         */
        public void add(int pushInt) {
            stackPush.push(pushInt);
        }
        
        /*
         * poll操作： 移除并返问队列头部的元素【注意队列为空的情况】
         * 1.stackPush  stackPop均为空时，队列为空队列，print 异常；
         * 2. 执行poll操作，即stackPush不为空时，将stackPush.pop的元素push进stackPop中【前提条件stackPop为空】
         * * 执行完“倒”的操作后，返回stackPop.pop()
         */
        public int poll() {
            if (stackPop.empty() && stackPush.empty()) {
                throw new RuntimeException("Queue is empty!");
            } else if (stackPop.empty()) {
                while (!stackPush.empty()) {
                    stackPop.push(stackPush.pop());
                }
            }
            return stackPop.pop();
        }

        /*
         * peek操作： 返回队列头部的元素  如果队列为空，则返回null【注意队列为空的情况】
         * 1.stackPush  stackPop均为空时，队列为空队列，print 异常；
         * 2. 执行peek操作，即stackPush不为空时，将stackPush.pop的元素push进stackPop中【前提条件stackPop为空】
         * 执行完“倒”的操作后，返回stackPop.peek()
         */
        public int peek() {
            if (stackPop.empty() && stackPush.empty()) {
                throw new RuntimeException("Queue is empty!");
            } else if (stackPop.empty()) {
                while (!stackPush.empty()) {
                    stackPop.push(stackPush.pop());
                }
            }
            return stackPop.peek();
        }
    }

    public static void main(String[] args) {
        TwoStacksQueue test = new TwoStacksQueue();
        test.add(1);
        test.add(2);
        test.add(3);
        System.out.println(test.peek());
        System.out.println(test.poll());
        System.out.println(test.peek());
        System.out.println(test.poll());
        System.out.println(test.peek());
        System.out.println(test.poll());
    }

}

运行结果：