Rhyme/剑指Offer Java 05 用两个栈实现一个队列Java版

剑指Offer Java 05 用两个栈实现一个队列Java版

package offer5;

import java.util.Stack;

/**
 * 用两个栈来实现一个队列，
 * 完成队列的push和pop操作
 * 
 * 思路:
 *  
 *  运用栈的特性后进先出
 *  那么如果我先将元素放入一个栈A中
 *  那么将栈A中的元素pop并push进栈B中
 *  那么在栈B执行pop操作，
 *  就和在队列中的到的结果相同了
 * 
 * @author RhymeChiang
 * @date 2018/01/15
 **/
public class Offer5<E> {

    private Stack<E> stackIn;
    private Stack<E> stackOut;

    {
        stackIn = new Stack<>();
        stackOut = new Stack<>();
    }

    /**
     * 模拟队列的push操作
     * <p>
     * 先将元素push进stackIn栈中
     * 在判断stackOut栈中是否有元素
     * <p>
     * 如果stackOut栈中有元素则不移动stackIn栈中的元素
     * 执行pop操作的时候，pop stackOut栈中的元素
     * <p>
     * 如果stackIn栈中没有元素则将stackIn中的所有元素pop,
     * 并push进stackOut栈中
     *
     * @param e
     */
    public void push(E e) {
        // 将元素先压入In栈
        stackIn.push(e);
        // Out栈中元素为空
        // 则将In栈中的元素全部pop
        // 并push进Out栈中
        if (stackOut.empty()) {
            while (!stackIn.empty()) {
                stackOut.push(stackIn.pop());
            }
        }
        // 如果Out栈中有元素，则不移动
    }

    /**
     * 模拟队列的pop操作
     *
     * @return
     */
    public E pop() {
        // 如果Out栈为空
        // 则将In栈的元素pop
        // 并push进Out栈中
        if (stackOut.empty()) {
            while (!stackIn.empty()) {
                stackOut.push(stackIn.pop());
            }
        }
        if (!stackOut.empty()) {
            return stackOut.pop();
        }
        return null;
    }


    public static void main(String[] args) {
        Offer5<Integer> offer5 = new Offer5<>();
        offer5.push(1);
        offer5.push(2);
        offer5.push(3);

        System.out.println(offer5.pop());
        System.out.println(offer5.pop());
        System.out.println(offer5.pop());
        System.out.println(offer5.pop());
    }
}

测试结果：