LeetCode Java刷题笔记—232. 用栈实现队列

232. 用栈实现队列

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）。

简单难度，使用两个栈即可实现，一个输入栈一个输出栈。

class MyQueue{

   //输入
   private Stack<Integer> in = new Stack<>();
   //输出
   private Stack<Integer> out = new Stack<>();

   public MyQueue(){

   }

   public void push( int x ){

      in.push( x );
   }

   public int pop(){
      // 如果输出栈为空，则将输入栈全部弹出并压入输出栈中，然后输出栈再出栈
      if( out.isEmpty() ){
         while( !in.isEmpty() ){
            out.push( in.pop() );
         }
      }
      return out.pop();
   }

   public int peek(){
      // 如果输出栈为空，则将输入栈全部弹出并压入输出栈中
      if( out.isEmpty() ){
         while( !in.isEmpty() ){
            out.push( in.pop() );
         }
      }
      return out.peek();
   }

   public boolean empty(){
      //判断输入输出栈是否都为空
      return in.isEmpty() && out.isEmpty();
   }
}