剑指offer系列5---两个栈 模拟队列的操作

【题目】两个栈 模拟队列的入队出队操作

package com.exe1.offer;

/**
 * [题目 ]：用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。
 * 队列的出队：应该是队列的列首先出，即栈底元素先出；思路：先把stack1元素一个个弹至上stack2中，再顺序出stack2中数即为队列的出队顺序。
 *      入队：先放到栈1中
 * @author WGS
 *
 */
import java.util.Stack;

public class CreateQueueWithTwoStacks {

    //1 创建两个栈
    Stack<Integer> stack1=new Stack<>();
    Stack<Integer> stack2=new Stack<>();
    //2 队列的入队操作
    public void push(int num){
        stack1.push(num);
    }
    //3 队列的出队操作
    public int pop(){
        if(stack2.size()<=0){
            while(stack1.size()>0){
                stack2.push(stack1.pop());//先把stack1中元素弹至stack2中，当stack1没有元素的时候，就将stack2元素弹出
            }
        }
        //异常处理
        if(stack2.size()==0){
            try{
                throw new Exception("empty queue");
            }catch(Exception e){
                e.printStackTrace();
            }
        }
        return stack2.pop();
    }

    
    
    
    
    public static void main(String[] args) {
        CreateQueueWithTwoStacks cq=new CreateQueueWithTwoStacks();
        cq.push(1);
        cq.push(2);
        cq.push(3);
        //模拟123顺序进队，下面可看到依然是以先进先出顺序123出
        System.out.println(cq.pop());
        System.out.println(cq.pop());
        System.out.println(cq.pop());

        //再模拟入队操作，再插入一个数
        cq.push(4);
        System.out.println(cq.pop());
        cq.push(5);
        System.out.println(cq.pop());
    }

}