CLRS10.1-6练习 - 用双栈实现队列

双栈实现队列算法：

分别考虑队列两种操作入队和出队，我们假设使用栈s1 s2，

s1用来模拟入队，s2用来模拟出队

入队：

入队操作直接执行s1.push即可

出队：

代码实现

 package hello;
 import java.util.*;

 public class TwoStackOneQueue<E> {
     private Stack<E> s1 = new Stack<>();
     private Stack<E> s2 = new Stack<>();

     public void enqueue(E item){
         s1.push(item);
     }

     public E dequeue(){
         if (s2.empty()){
             if(s1.empty()){
                 throw new ArrayIndexOutOfBoundsException();
             }else{
                 popS1ToS2();
                 return s2.pop();
             }
         }else{
             return s2.pop();
         }
     }

     private void popS1ToS2(){
         while(!s1.empty()){
             s2.push(s1.pop());
         }
     }

     public static void main(String[] args){
         TwoStackOneQueue<Integer> tsoq = new TwoStackOneQueue<>();
         for (int i = 0; i < 20; i++){
             tsoq.enqueue(i);
         }
         for (int i = 0; i < 10; i++) {
             System.out.println(tsoq.dequeue());
         }
         for (int i = 20; i < 40; i++){
             tsoq.enqueue(i);
         }
         for (int i = 0; i < 30; i++) {
             System.out.println(tsoq.dequeue());
         }
     }
 }