用两个栈实现队列，支持队列的基本操作。

描述

用两个栈实现队列，支持队列的基本操作。

输入描述：

第一行输入一个整数N，表示对队列进行的操作总数。

下面N行每行输入一个字符串S，表示操作的种类。

如果S为"add"，则后面还有一个整数X表示向队列尾部加入整数X。

如果S为"poll"，则表示弹出队列头部操作。

如果S为"peek"，则表示询问当前队列中头部元素是多少。

输出描述：

对于每一个为"peek"的操作，输出一行表示当前队列中头部元素是多少。

示例1

输入：

6
add 1
add 2
add 3
peek
poll
peek

输出：

1
2

使用两个栈，入队栈和出队栈；入队使用入队栈，出队使用出队栈；

入队栈元素要转移到出队栈，必须保证： 

1.转移元素前出队栈为空 

2.转移元素后入队栈为空

 

(1) 入队操作时元素直接进入入队栈；

(2) 出队操作时先看出队栈中是否有元素，如果有就先把其中的元素弹栈，否则就先将入队栈中的元素全部弹出到出队栈，再对出队栈进行弹栈出队，这样就能够保证先进先出。

(3) 查看队首元素与(2)的操作类似，只是不将元素弹出，仅仅只是查看栈顶元素。

 

import java.util.Scanner;
import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        TowStacksQueue myQueue=new TowStacksQueue();
        Scanner in = new Scanner(System.in);
        int N=in.nextInt();//输入操作数N
        for(int i=0;i<=N;i++){//进行N次输入
            String S=in.nextLine();//进行N次输入，每次输入的S都进行下面的三个判断
            if(S.contains("add")){
                Integer newNum=new Integer(S.split(" ")[1]);
                myQueue.add(newNum);
            }
            if(S.equals("peek")){
                int min = myQueue.peek();
                System.out.println(min);
            }
            if(S.equals("poll")){
                int min = myQueue.poll();
            }
        }
    }
}    
class TowStacksQueue{
    public Stack<Integer> stackPush;//从外部压入的栈....入队栈
    public Stack<Integer> stackPop;//取出元素的栈....出队栈
    
    public TowStacksQueue(){
        stackPush=new Stack<Integer>();
        stackPop =new Stack<Integer>();
    }
    
    private void pushToPop(){
        if(stackPop.empty()){//只有当出队栈是空的情况下，才从入队栈里压入元素
            while(!stackPush.empty()){//只要入队栈有元素都要一次性压入 出队栈
                stackPop.push(stackPush.pop());//依次压入 出队栈
            }
        }
    }
    
    public void add(int pushInt){
        stackPush.push(pushInt);//从外部压入 入队栈
        pushToPop();//每压入一个元素，都要去尝试一下是否能压入 出队栈
    }
    
    public int poll(){
        if(stackPush.empty() && stackPop.empty()){
            throw new RuntimeException("你的队列里面啥也没有！");
        }
        pushToPop();//每次执行取出操作，都先尝试一下是否能够压入出队栈  保证出队栈内有元素
        return stackPop.pop();
    }
    
    public int peek(){
        if(stackPush.empty() && stackPop.empty()){
            throw new RuntimeException("你的队列里面啥也没有！");
        }
        pushToPop();
        return stackPop.peek();
    }
}