由两个栈组成队列

【题目】

             编写一个类，用两个栈实现队列，支持队列的基本操作（add、poll、peek）。

【解答】​

            栈的特点是先进后出，而队列的特点是先进先出。我们用两个栈正好能把顺序反过来实现类似队列的操作。

            具体实现上，是把一个栈作为压入栈，在压入数据时只往这个栈中压入，记为stackPush；另一个栈只作为弹出栈，在弹出数据时只从这个栈中弹出，记为stackPop。

            因为数据压入栈的时候，顺序是先进先出的。那么只要把stackPush的数据再次压入stackPop中，顺序就变回来了。

将1-5依次压入stackPush，那么从stackPush的栈顶到栈底为5~1，此时依次再将5~1倒入stackPop，那么从stackPop的栈顶到栈底就变成了1~5.再从stackPop中弹出，顺序就和队列一样了。​

            BUT，要做到以下两点：

           （1）如果stackPush要往stackPop中压入数据，那么必须一次性把stackPush中的数据全部压入。

            （2）如果stackPop不为空，stackPush绝对不能向stackPop中压入数据。

            违反以上两点都会发生错误。

【代码】

TwoStackQueue.java：

package cn.hl.p2;

import java.util.Stack;
/**
 * 题目：由两个栈组成的队列
 * 要求：用两个栈实现队列，支持队列的基本操作（add、poll、peek）
 * @author 猩生柯北
 *
 */
public class TwoStackQueue {
    public Stack<Integer> stackPush;
    public Stack<Integer> stackPop;
    
    public TwoStackQueue(){
        stackPush = new Stack<Integer>();
        stackPop = new Stack<Integer>();
    }
    
    public void add(int pushInt){
        stackPush.push(pushInt);
    }
    
    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();
    }
    
    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();
    }
}

Test.java：

package cn.hl.p2;

public class Test {
    public static void main(String[] args) {
        TwoStackQueue tsq = new TwoStackQueue();
        tsq.add(3);
        tsq.add(8);
        tsq.add(-5);
        tsq.add(0);
        tsq.add(8);
        System.out.print("输出队列的head元素：");
        System.out.println(tsq.peek());
        System.out.println("===========");
        System.out.println(tsq.poll());
    }

}

【运行结果】