算法与数据机构学习_第一章.栈和队列_2.实现2个栈组成的队列

2个栈组成的队列

===========

算法要求：用2个栈实现队列的基本操作（队列的先进先出原则）

思路：栈是先进后出的原则，二队列是先进先出原则。栈的出栈顺序和队列刚好是相反的，因此，可以用2个栈来把顺序反过来，实现先进先出的顺序。

算法过程：定义2个栈，stack1和stack2，可以往stack1中push数据，每次每次加入数据都是往stack1中加入数据，而要peek或pup数据时，从stack2里面取，这样的话，就要实现在pup或peek时，将stack1中的数据压栈到stack2中，此时，stack2中的元素顺序是先进先出的，只要访问stack2的pup和peek方法返回对应的元素即可。

public class QueueFromStack {
    public static void main(String[] args) {

        StackQueue<Integer> stackQueue = new StackQueue();
        stackQueue.add(1);
        stackQueue.add(2);
        stackQueue.add(3);
        stackQueue.add(4);
        System.out.println(stackQueue);

        Integer peek = stackQueue.peek();
        System.out.println(peek);

        Integer pop = stackQueue.pop();
        System.out.println(pop);
    }
}

class StackQueue<E> {
    Stack<E> stack1 = new Stack<>(); //存放第一次加入的数据
    Stack<E> stack2 = new Stack<>(); //存放移到另一个栈的中数据

    public void add(E e) {
        stack1.push(e);
    }

    public E peek() {
        if (stack1.empty() && stack2.empty()) { //都为空，则数据为空
            throw new RuntimeException("stack is empty !");
        }
        else {
            if (stack2.empty()) {
                while(!stack1.empty()){
                    stack2.push(stack1.pop());
                }
            }
        }
        return stack2.peek();
    }

    public E pop() {
        if (stack1.empty() && stack2.empty()) { //都为空，则数据为空
            throw new RuntimeException("stack is empty !");
        }
        else {
            if (stack2.empty()) {
                while(!stack1.empty()){
                    stack2.push(stack1.pop());
                }
            }
        }
        return stack2.pop();
    }

}