两个栈实现一个队列

Java用两个栈实现队列

import java.util.Stack;
import org.junit.Test;
public class QueueByTwoStack {
    private Stack<Integer> stack1=new Stack<Integer>();
    private Stack<Integer> stack2=new Stack<Integer>();

    public QueueByTwoStack() {
        super();
    }
    /**
     * 入队
     * @param ele 要入队的元素
     */
    private void push(int ele){
        stack1.push(ele);
    }

    /**
     * 出队
     * @return 队首元素
     */
    public int pop(){
        if(stack1.isEmpty() && stack2.isEmpty()){//如果两个栈都空了，说明队列空
            try {
                throw new Exception("队列空");
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        while(stack2.size()<=0){//如果栈2中没有元素
            while(stack1.size()>0){//如果栈1中有元素
                stack2.push(stack1.pop());//栈1出栈，栈2入栈，为了满足队列先进先出
            }
        }

        return stack2.pop();
    }
    /**
     * 判断是否队空
     * @return
     */
    private boolean isEmpty(){
        if(stack1.isEmpty() && stack2.isEmpty()){//如果两个栈都空了，说明队列空
            return true;
        }else{
            return false;
        }
    }
    /**
     *查看队首元素
     * @return 队首元素
     */
    public int peek(){
        if(isEmpty()){
            try {
                throw new Exception("队列空");
            } catch (Exception e) {
                e.printStackTrace();
            }
        }else{
            while(stack2.isEmpty()){//如果栈2中没有元素
                while(!stack1.isEmpty()){//如果栈1中有元素
                    stack2.push(stack1.pop());//栈1出栈，栈2入栈，为了满足队列先进先出
                }
            }
        }
        return stack2.peek();
    }

    @Test
    public void testQueueByTwoStack(){

        QueueByTwoStack q=new QueueByTwoStack();
        for (int i = 0; i <10; i++) {
            q.push(i);
        }

        System.out.println("队列出队序列：");
        while(!q.isEmpty()){
            System.out.print(q.pop()+" ");
        }
    }
}