程序员算法指南-设计一个由两个栈组成的队列(自带详细注释)

【说明】：本文来自左程云老师所著的《程序员面试代码指南》
【题目】编写 一个类，用两个栈实现队列，支持队列的基本操作（add, pol,peek）
【解答】栈的特点是先进后出，而队列的特点是先进先出。
我们用两个栈正好能把顺序反过来实现类似队列的操作。
具体的做法是一个栈作为压入栈，在压入数据的时候只往这个栈压入，记为 stackPush；另一个栈只作为弹出栈，在弹出数据时候只从这个栈弹出，记为stackPop。
因为数据压入栈的时候， 顺序是先进后出的。那么只要把stackPush的数据再压入到stackPop中，顺序就变回来了。
注意两点：
1.如果stackPush要往stackPop中压入数据，那么必须一次性把stackPush中的数据全部压入。
2.如果stackPop不为空，stackPush决不能向stackPop中压入数据。

import java.util.Stack;

public class MyStack {
    private Stack<Integer> stackPush;//创建一个栈的引用用于保存正常栈
    private Stack<Integer> stackPop;//再创建一个栈的引用用于保存倒置栈

    public MyStack() {
        this.stackPush = new Stack<>();//创建一个栈用于保存正常栈
        this.stackPop = new Stack<>();//再创建一个栈用于保存倒置栈
    }

    public void add(int pushInt){//让pushInt加入队列
        stackPush.push(pushInt);//先放置到正常栈中
    }

    public int poll(){//将正常栈中的数倒置到-倒置栈中
        if(stackPush.empty()&&stackPop.empty()){//如果没有数据存入，抛出异常
            throw new RuntimeException("Queue is empty!");
        }else if(stackPop.empty()){//如果此时倒置栈没有数据，则将正常栈中数据倒置到-倒置栈
                while(!stackPush.empty()){//直到正常栈无数据
                    stackPop.push(stackPush.pop());//正常栈抛出，利用返回值倒置栈进入
                }
        }
        return stackPop.pop();//返回最后一个倒置的数，即为队列的头
    }

    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 class Demo {
    public static void main(String[] args) {
        MyStack stack = new MyStack();
        stack.add(3);
        stack.add(2);
        stack.add(1);
        System.out.println(stack.poll());//3
        stack.add(4);
        stack.add(5);
        System.out.println(stack.poll());//2
        System.out.println(stack.poll());//1
        System.out.println(stack.poll());//4
        System.out.println(stack.peek());//5
        System.out.println(stack.poll());//5
    }
}